From 6aa7d0b8ee7359da0e7180590de411f1b63d578a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 4 Oct 2023 11:41:21 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 17 ++++++++---------
1 file changed, 8 insertions(+), 9 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 9a4681b2..905e3958 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -231,18 +231,17 @@ class Trainer(object):
l1_loss = utils.pixel_loss(image_hr, fake)
logging.debug("Calculated Pixel Loss")
- # loss_RaG = ra_gen(image_hr, fake)
- # logging.debug("Calculated Relativistic"
- # "Average (RA) Loss for Generator")
- #
- # disc_loss = ra_disc(image_hr, fake)
- # logging.debug("Calculated RA Loss Discriminator")
- disc_loss = l1_loss
+ loss_RaG = ra_gen(image_hr, fake)
+ logging.debug("Calculated Relativistic"
+ "Average (RA) Loss for Generator")
+
+ disc_loss = ra_disc(image_hr, fake)
+ logging.debug("Calculated RA Loss Discriminator")
+
# TDR, we don't have percep_loss
# gen_loss = percep_loss + lambda_ * loss_RaG + eta * l1_loss
- # gen_loss = lambda_ * loss_RaG + eta * l1_loss
- gen_loss = l1_loss
+ gen_loss = lambda_ * loss_RaG + eta * l1_loss
logging.debug("Calculated Generator Loss")
disc_metric(disc_loss)
--
GitLab