From 4d8b92486f9ff83e22b63e520803d348b36556c7 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 12 Oct 2023 15:02:10 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 8a625e92..d1d1fbcf 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -71,6 +71,7 @@ class Trainer(object):
total_steps = phase_args["num_steps"]
metric = tf.keras.metrics.Mean()
psnr_metric = tf.keras.metrics.Mean()
+ mse_metric = tf.keras.metrics.Mean()
# Generator Optimizer
G_optimizer = tf.optimizers.Adam(
@@ -117,6 +118,7 @@ class Trainer(object):
for epoch in range(NUM_WU_EPOCHS):
print('start epoch #: ', epoch)
metric.reset_states()
+ psnr_metric.reset_states()
for image_lr, image_hr in self.dataset:
num_steps = train_step(image_lr, image_hr)
--
GitLab