diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 9cc86e8cfb980b7bb10078d6e13fb13a99173da0..89cbdfadf7680f29107c5c153e2e76ce5e3d471a 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -29,7 +29,7 @@ TRACK_MOVING_AVERAGE = False EARLY_STOP = True NOISE_TRAINING = False -NOISE_STDDEV = 0.10 +NOISE_STDDEV = 0.01 DO_AUGMENT = True DO_ZERO_OUT = False @@ -218,9 +218,9 @@ class SRCNN: add_noise = None noise_scale = None - if is_training: + if is_training and NOISE_TRAINING: add_noise = True - noise_scale = 0.005 + noise_scale = NOISE_STDDEV data_norm = [] for param in data_params: