diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index cb8b00e8ba1c37cbd7563ef44b2a0fa59310fec1..7e58b4321b3e7ae52028a24ea6805a9d72f14631 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)