remove assert, probably not necessary

2022-09-25 09:32:19 +00:00
parent dcda03b942
commit 26916c00e7
1 changed files with 2 additions and 2 deletions
@@ -136,8 +136,8 @@ def train():


 def test():
-    load_model(model.encoder, 'checkpoints/' + config['name'] + '_enc', assert_consumed_on=True)
-    load_model(model.decoder, 'checkpoints/' + config['name'] + '_dec', assert_consumed_on=True)
+    load_model(model.encoder, 'checkpoints/' + config['name'] + '_enc')
+    load_model(model.decoder, 'checkpoints/' + config['name'] + '_dec')
    for batch in dataset_train:
        batch = static_dataset_train.expand_batch(batch_to_dict(batch, config['signature_dynamic']))