diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index aea68231617991bf85502043a85b784718045bc5..d3706f0f648ca5e23fe64e54d913ce64d9fd34a5 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -185,6 +185,7 @@ class Trainer(object):
         G_optimizer=G_optimizer,
         D=discriminator,
         D_optimizer=D_optimizer)
+    ckpt_manager = tf.train.CheckpointManager(checkpoint, self.model_dir, max_to_keep=7)
     if not tf.io.gfile.exists(
         os.path.join(
             self.model_dir,