From 78a650b29b4df5ae8e15c404f6eb617e71b6c6f3 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Tue, 25 Apr 2023 15:07:35 -0500 Subject: [PATCH] bug fixes ... --- data/gan.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 = []