Skip to content
Snippets Groups Projects
Commit 0fa8c7c9 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 8e998271
Branches
No related tags found
No related merge requests found
......@@ -694,9 +694,9 @@ class SRCNN:
self.build_training()
self.build_evaluation()
ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
ckpt.restore(ckpt_manager.latest_checkpoint)
# ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
# ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
# ckpt.restore(ckpt_manager.latest_checkpoint)
def do_inference(self, inputs):
self.reset_test_metrics()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment