From f65f6625e8d524718b004fe53d4665ab5c2cac66 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 6 Oct 2023 10:02:32 -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 173841ab..7467b257 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -92,6 +92,7 @@ class Trainer(object):
with tf.GradientTape() as tape:
fake = generator.unsigned_call(image_lr)
loss = utils.pixel_loss(image_hr, fake) * (1.0 / self.batch_size)
+ # loss = utils.pixel_loss_mse(image_hr, fake) * (1.0 / self.batch_size)
psnr_metric(tf.reduce_mean(tf.image.psnr(fake, image_hr, max_val=256.0)))
# gen_vars = list(set(generator.trainable_variables))
gen_vars = generator.trainable_variables
--
GitLab