From 3fd6818ef76ae9ca93bac56717bfe438f984899d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 20 Apr 2021 14:22:27 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index cb8b00e8..7e58b432 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -18,7 +18,7 @@ PROC_BATCH_SIZE = 10240
PROC_BATCH_BUFFER_SIZE = 50000
NumLabels = 1
BATCH_SIZE = 256
-NUM_EPOCHS = 20
+NUM_EPOCHS = 100
TRACK_MOVING_AVERAGE = False
@@ -377,7 +377,7 @@ class IcingIntensityNN:
#self.loss = tf.keras.losses.SparseCategoricalCrossentropy() # For multi-class
# decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps)
- initial_learning_rate = 0.0016
+ initial_learning_rate = 0.002
decay_rate = 0.95
steps_per_epoch = int(self.num_data_samples/BATCH_SIZE) # one epoch
# decay_steps = int(steps_per_epoch / 2)
--
GitLab