From be74e20a8f1569bf8ba575323d7b03fc9ec6994b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 30 Apr 2021 10:38:47 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 40af66f3..98f27613 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1,5 +1,5 @@
import tensorflow as tf
-#import tensorflow_addons as tfa
+import tensorflow_addons as tfa
from util.setup import logdir, modeldir, cachepath
from util.util import homedir
@@ -18,7 +18,7 @@ PROC_BATCH_SIZE = 2046
PROC_BATCH_BUFFER_SIZE = 50000
NumLabels = 1
BATCH_SIZE = 256
-NUM_EPOCHS = 80
+NUM_EPOCHS = 50
TRACK_MOVING_AVERAGE = False
@@ -393,7 +393,7 @@ class IcingIntensityNN:
self.learningRateSchedule = tf.keras.optimizers.schedules.ExponentialDecay(initial_learning_rate, decay_steps, decay_rate)
optimizer = tf.keras.optimizers.Adam(learning_rate=self.learningRateSchedule)
- # optimizer = tfa.optimizers.MovingAverage(optimizer)
+ optimizer = tfa.optimizers.MovingAverage(optimizer)
if TRACK_MOVING_AVERAGE:
ema = tf.train.ExponentialMovingAverage(decay=0.999)
--
GitLab