snapshot...

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()