diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 136de6fe0efb91c638d2d894cdf0bcd86cff802a..76d242766758231031231e766049c64ea765dc81 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -510,6 +510,7 @@ class SRCNN:
         else:
             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.writer_train = tf.summary.create_file_writer(os.path.join(logdir, 'plot_train'))
         self.writer_valid = tf.summary.create_file_writer(os.path.join(logdir, 'plot_valid'))