diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index 5207706779baf267ec873124e54e47261828dd5b..77eb9ad89c96c29af59cb0bbac9b64a05382fdc8 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/modules/GSOC/E2_ESRGAN/lib/train.py @@ -12,7 +12,7 @@ from pathlib import Path logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG) NUM_WU_EPOCHS = 30 -NUM_EPOCHS = 20 +NUM_EPOCHS = 30 class Trainer(object):