From e4ff7c83367b02f7fcb924c897346ba226df478a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 28 Sep 2023 14:58:04 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 1372e380..9a4681b2 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -231,15 +231,18 @@ 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")
+ # 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
- 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 = lambda_ * loss_RaG + eta * l1_loss
+ gen_loss = l1_loss
logging.debug("Calculated Generator Loss")
disc_metric(disc_loss)
--
GitLab