diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 6b09eaf7c64811bc3327c7a1fb888fddca898fe6..7b047f00af4b798dcdc2fe18e9abeb099e62f412 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -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()