diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 76d242766758231031231e766049c64ea765dc81..b0678915f604a68606ba9c41ab30cd5d37451357 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -554,9 +554,6 @@ class SRCNN: for mini_batch_test in tst_ds: self.test_step(mini_batch_test) - # if NumClasses == 2: - # f1, mcc = self.get_metrics() - with self.writer_valid.as_default(): tf.summary.scalar('loss_val', self.test_loss.result(), step=step) tf.summary.scalar('acc_val', self.test_accuracy.result(), step=step) @@ -614,7 +611,6 @@ class SRCNN: 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) self.reset_test_metrics()