From 95d07f909b8788654711d63eb0ed60e7ff3d25b4 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 12 Oct 2023 10:43:22 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 25d727cb..d4b827d6 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -15,6 +15,7 @@ NUM_WU_EPOCHS = 14
NUM_EPOCHS = 20
num_chans = 2
PSNR_MAX = 256.0
+GENERATOR_ONLY = True
class Trainer(object):
@@ -153,6 +154,8 @@ class Trainer(object):
utils.save_checkpoint(checkpoint, "phase_1", self.model_dir)
previous_loss = psnr_metric.result()
start_time = time.time()
+ if GENERATOR_ONLY:
+ return
def train_gan(self, generator, discriminator):
""" Implements Training routine for ESRGAN
--
GitLab