Skip to content
Snippets Groups Projects
Commit 07af94fc authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 4c515a71
No related merge requests found
Hide whitespace changes
Inline Side-by-side
modules/GSOC/E2_ESRGAN/lib/train.py
+ 1
− 1
View file @ 07af94fc
...@@ -14,7 +14,7 @@ logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging ...@@ -14,7 +14,7 @@ logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging
NUM_WU_EPOCHS = 1 NUM_WU_EPOCHS = 1
NUM_EPOCHS = 20 NUM_EPOCHS = 20
num_chans = 2 num_chans = 2
PSNR_MAX = 1.0 PSNR_MAX = 256.0
class Trainer(object): class Trainer(object):
""" Trainer class for ESRGAN """ """ Trainer class for ESRGAN """
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment