From 33d54fc3c1c26252b0a6f6feeae01e061fe3e3e6 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 8 Sep 2021 10:21:44 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 9b6f247d..d4c98087 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -240,7 +240,7 @@ class IcingIntensityNN:
nda = normalize(nda, param, mean_std_dct, add_noise=True, noise_scale=0.01, seed=42)
else:
nda = normalize(nda, param, mean_std_dct)
- if DO_ZERO_OUT:
+ if DO_ZERO_OUT and is_training:
try:
zero_out_params.index(param)
nda[:,] = 0.0
--
GitLab