diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 97e079ab130236109722b45e16da27e084cb1087..6c32384a9f1a1a60f4d931badbfb20acd9372cf0 100644
--- a/modules/deeplearning/espcn.py
+++ b/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)