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

minor...

parent 7abcf11a
Branches
No related tags found
No related merge requests found
......@@ -740,7 +740,7 @@ class ESPCN:
self.test_preds = preds
def do_evaluate(self, nda_lr, ckpt_dir, prob_thresh=0.5):
def do_evaluate(self, nda_lr, ckpt_dir):
ckpt = tf.train.Checkpoint(step=tf.Variable(1), model=self.model)
ckpt_manager = tf.train.CheckpointManager(ckpt, ckpt_dir, max_to_keep=3)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment