From 374ac36b8fe2fe533daafd0947b6371351c5199f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 4 Oct 2023 12:19:33 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index aea68231..d3706f0f 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,
--
GitLab