From 151015e9f93b72d192294720e7e599291edfbc05 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 12 Oct 2023 10:46:01 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index d4b827d6..8a625e92 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -154,8 +154,6 @@ 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
@@ -342,3 +340,5 @@ class Trainer(object):
last_psnr = psnr_metric.result()
utils.save_checkpoint(checkpoint, "phase_2", self.model_dir)
start = time.time()
+ if GENERATOR_ONLY:
+ return
--
GitLab