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

more logging

Browse Source
This commit is contained in:
luni
2022-10-16 14:54:33 +00:00
parent 3dac84e189
commit 3939338799
9 changed files with 22 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
auto_encoder_curv.py
+2 -2
View File
@@ -211,9 +211,9 @@ if __name__ == '__main__':
optimizer = tf.optimizers.Adam(learning_rate=config['lr'])
ckpt_enc = tf.train.Checkpoint(model=model.encoder)
manager_enc = tf.train.CheckpointManager(ckpt_enc, 'checkpoints/' + config['name'] + '_enc', max_to_keep=3)
manager_enc = tf.train.CheckpointManager(ckpt_enc, 'checkpoints/' + config['name'] + '_enc', max_to_keep=20)
ckpt_dec = tf.train.Checkpoint(model=model.decoder)
manager_dec = tf.train.CheckpointManager(ckpt_dec, 'checkpoints/' + config['name'] + '_dec', max_to_keep=3)
manager_dec = tf.train.CheckpointManager(ckpt_dec, 'checkpoints/' + config['name'] + '_dec', max_to_keep=20)
if testing:
test()