diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index c2539a9ae4a59911fa1ffd4f5b9476cd0c899b64..9002f2902b2a5151e31dd00c32e6a8300b4852f3 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -219,7 +219,7 @@ class IcingIntensityFCN: self.X_img = tf.keras.Input(shape=(None, None, n_chans)) self.inputs.append(self.X_img) - self.inputs.append(tf.keras.Input(None, None, 5)) + self.inputs.append(tf.keras.Input(shape=(None, None, 5))) self.flight_level = 0