diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 59d37ec432042e980858dbc63ccdd95655f4e9c9..32465e67bca90a7aa498cf647b62ed5dd45180db 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -970,6 +970,7 @@ class IcingIntensityNN: # with tf.device('/device:GPU:'+str(self.gpu_device)): self.setup_pipeline(filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst) self.build_model() + self.model.summary() self.build_training() self.build_evaluation() self.do_training()