diff --git a/modules/deeplearning/icing_only_fcn.py b/modules/deeplearning/icing_only_fcn.py index e31328a84448e15ade24c5849599ab62f7d64400..e79165b7d7b1046f8bdec551ea6bf0c88ac34ffa 100644 --- a/modules/deeplearning/icing_only_fcn.py +++ b/modules/deeplearning/icing_only_fcn.py @@ -20,8 +20,8 @@ if NumClasses == 2: else: NumLogits = NumClasses -BATCH_SIZE = 128 -NUM_EPOCHS = 50 +BATCH_SIZE = 64 +NUM_EPOCHS = 60 TRACK_MOVING_AVERAGE = False EARLY_STOP = True @@ -721,7 +721,7 @@ class IcingIntensityFCN: self.loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False) # For multi-class # decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps) - initial_learning_rate = 0.002 + initial_learning_rate = 0.001 decay_rate = 0.95 steps_per_epoch = int(self.num_data_samples/BATCH_SIZE) # one epoch decay_steps = int(steps_per_epoch)