diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 5dd8219ff51fbdffc876e6f1f8ea5a16636f644e..a20c5c48fa71b973194d507a8c1dafa55440d8cd 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -418,7 +418,7 @@ class IcingIntensityNN: momentum = 0.99 num_filters = 16 - # num_filters = 10 + # num_filters = 12 conv = tf.keras.layers.Conv2D(num_filters, 5, strides=[1, 1], padding=padding, activation=activation)(self.inputs[0]) conv = tf.keras.layers.MaxPool2D(padding=padding)(conv) @@ -861,7 +861,7 @@ class IcingIntensityNN: self.restore(ckpt_dir) def run_evaluate(self, filename, ckpt_dir): - data_dct, ll, cc = make_for_full_domain_predict(filename) + data_dct, ll, cc = make_for_full_domain_predict(filename, name_list=train_params) self.setup_eval_pipeline(data_dct, len(ll)) self.build_model() self.build_training()