luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove assert, probably not necessary

Browse Source
This commit is contained in:
luni
2022-09-25 09:32:19 +00:00
parent dcda03b942
commit 26916c00e7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
auto_encoder_static.py
+2 -2
View File
@@ -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']))