From 26916c00e7df9046700ed99d21aa750987adb7f1 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sun, 25 Sep 2022 09:32:19 +0000 Subject: [PATCH] remove assert, probably not necessary --- auto_encoder_static.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/auto_encoder_static.py b/auto_encoder_static.py index 7c14d39..0b22efc 100644 --- a/auto_encoder_static.py +++ b/auto_encoder_static.py @@ -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']))