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