From ab7f6e6357180ca8dc2ff9fdb3587652377c4ccd Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 26 Aug 2022 10:26:52 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/espcn.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 97e079ab..6c32384a 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)
-- 
GitLab