From f42837f61a2c1cc91e85a44d843c4d75ad091616 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 22 Aug 2023 12:51:19 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 9e130450..f9d56424 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -66,7 +66,6 @@ class Trainer(object):
decay_step = decay_params["step"]
decay_factor = decay_params["factor"]
total_steps = phase_args["num_steps"]
- print('total number of steps: ', total_steps)
metric = tf.keras.metrics.Mean()
psnr_metric = tf.keras.metrics.Mean()
@@ -79,6 +78,7 @@ class Trainer(object):
checkpoint = tf.train.Checkpoint(
G=generator,
G_optimizer=G_optimizer)
+ ckpt_manager = tf.train.CheckpointManager(checkpoint, self.model_dir, max_to_keep=7)
status = utils.load_checkpoint(checkpoint, "phase_1", self.model_dir)
logging.debug("phase_1 status object: {}".format(status))
--
GitLab