backup

2022-10-04 21:50:58 +00:00
parent 5b82041acb
commit 3c798b473d
8 changed files with 270 additions and 125 deletions
@@ -214,8 +214,8 @@ def load_model(model, checkpoint, disable_assert=False, assert_consumed_on=False
    ckpt = tf.train.Checkpoint(model=model)
    manager = tf.train.CheckpointManager(ckpt, checkpoint, max_to_keep=1)
    if not disable_assert:
-        assert manager.latest_checkpoint is not None
-        assert manager.latest_checkpoint.split('/')[-1] != 'ckpt-0'
+        assert manager.latest_checkpoint is not None, checkpoint
+        assert manager.latest_checkpoint.split('/')[-1] != 'ckpt-0', checkpoint
    if assert_consumed_on:
        ckpt.restore(manager.latest_checkpoint).assert_consumed()
    else: