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
cvae_curv.py
+1 -1
View File
@@ -388,7 +388,7 @@ if __name__ == '__main__':
ckpt = tf.train.Checkpoint(model=model)
manager = tf.train.CheckpointManager(ckpt, 'checkpoints/' + name, max_to_keep=3)
manager = tf.train.CheckpointManager(ckpt, 'checkpoints/' + name, max_to_keep=20)
step = 0
for batch in dataset_train: