diff --git a/data/gan.py b/data/gan.py
index a794ebe..7860e0f 100644
--- a/data/gan.py
+++ b/data/gan.py
@@ -667,7 +667,9 @@ class Predict(GNet):
 
                 fake    = self.generator.network(inputs=z, label=label)
                 init    = tf.compat.v1.global_variables_initializer()
-                saver   = tf.compat.v1.train.Saver()
+                print ([self.CHECKPOINTS])
+                # saver   = tf.compat.v1.train.Saver()
+                saver = tf.compat.v1.train.Saver(max_to_keep=len(self.CHECKPOINTS))
                 df              = pd.DataFrame()
                 CANDIDATE_COUNT = args['candidates'] if 'candidates' in  args else 1 #0 if self.ROW_COUNT < 1000 else 100
                 candidates = []