From 77bd8f931757d2983909c8ca11dff0d240c44f92 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 21 Aug 2023 16:22:05 -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 b1278ab1..fc992c7c 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -63,6 +63,7 @@ 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()
--
GitLab