diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 9b6f247d033a0a4a79f39fbb280e5021e95d538f..d4c98087002b2ec0adb08d25c9f0f6f18c1b7fb4 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