diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index f8fd9b6885e7fc1b15cb40dc682217877144d0fc..3d041bbbce0a4e7b9b6fa1adda4bdece7da8b976 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -592,7 +592,7 @@ class IcingIntensityFCN: # activation = tf.nn.elu activation = tf.nn.leaky_relu - num_filters = len(self.train_params) * 4 + num_filters = len(self.train_params) * 8 input_2d = self.inputs[0] @@ -731,7 +731,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.0005 + 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)