From e30154ea8462b6efc8ef0b0dace04b91b01a2433 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 9 Oct 2023 11:01:37 -0500 Subject: [PATCH] snapshot... --- modules/GSOC/E2_ESRGAN/lib/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index a53b1669..5b21942b 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/modules/GSOC/E2_ESRGAN/lib/train.py @@ -11,11 +11,12 @@ from pathlib import Path logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG) -NUM_WU_EPOCHS = 1 +NUM_WU_EPOCHS = 10 NUM_EPOCHS = 20 num_chans = 2 PSNR_MAX = 256.0 + class Trainer(object): """ Trainer class for ESRGAN """ -- GitLab