snapshot...

modules/deeplearning/espcn.py

@@ -225,10 +225,12 @@ class ESPCN:
         # if label_param != 'cloud_fraction':
         #     label = scale(label, label_param, mean_std_dct)
 
+        add_noise = None
+        noise_scale = None
         if is_training:
-            data = normalize(data, data_param, mean_std_dct, add_noise=True, noise_scale=0.005)
-        else:
-            data = normalize(data, data_param, mean_std_dct)
+            add_noise = True
+            noise_scale = 0.005
+        data = normalize(data, data_param, mean_std_dct, add_noise=add_noise, noise_scale=noise_scale)
 
         if label_param != 'cloud_fraction':
             label = normalize(label, label_param, mean_std_dct)