more logging

2022-10-16 14:54:33 +00:00
parent 3dac84e189
commit 3939338799
9 changed files with 22 additions and 21 deletions
@@ -388,7 +388,7 @@ if __name__ == '__main__':
    test()

    # ckpt = tf.train.Checkpoint(model=model)
-    # manager = tf.train.CheckpointManager(ckpt, 'checkpoints/' + config['name'], max_to_keep=3)
+    # manager = tf.train.CheckpointManager(ckpt, 'checkpoints/' + config['name'], max_to_keep=20)

    # valid_iterator = iter(dataset_valid)
    # step = 0