snapshot...

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)