diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py index 80e71fc..c3cf25a 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_mono.py +++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py @@ -36,6 +36,7 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext arch_name = model_arch_enc )(layer_input) + print("ENCODER output_shape", layer_next.shape) # BOTTLENECK layer_next = tf.keras.layers.Dense(name="cns.stage.bottleneck.dense2", units=feature_dim)(layer_input)