From fd44a03a422ff6d98f62cc475239b904469556b4 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 13 Nov 2022 11:57:00 -0600
Subject: [PATCH] snapshot...

---
 modules/deeplearning/icing_fcn.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 5052bdf7..ccceb7de 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -571,12 +571,13 @@ class IcingIntensityFCN:
         num_filters = len(self.train_params) * 16
 
         input_2d = self.inputs[0]
-        conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=None)(input_2d)
-        print(conv.shape)
-        skip = conv
 
         if NOISE_TRAINING:
-            conv = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv)
+            conv = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(input_2d)
+
+        conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=activation)(conv)
+        print(conv.shape)
+        skip = conv
 
         conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=activation)(conv)
         conv = tf.keras.layers.MaxPool2D(padding=padding)(conv)
-- 
GitLab