From 2488fbb7ca8be9bd9b05d3c7ff690ce2ac8b1874 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 22 Aug 2023 11:40:01 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/train.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index a0498295..9e130450 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -109,6 +109,7 @@ class Trainer(object):
return distributed_metric
for epoch in range(NUM_WU_EPOCHS):
+ print('start epoch #: ', epoch)
for image_lr, image_hr in self.dataset:
num_steps = train_step(image_lr, image_hr)
@@ -263,6 +264,7 @@ class Trainer(object):
last_psnr = 0
for epoch in range(NUM_EPOCHS):
+ print('start epoch #: ', epoch)
for image_lr, image_hr in self.dataset:
num_step = train_step(image_lr, image_hr)
--
GitLab