bug fixes ...

2 years ago · 78a650b29b
parent e5af702ddb
commit 78a650b29b
1 changed files with 3 additions and 1 deletions
--- 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 = []