From d0dbc50bb78c5bb1a35ef1b09418b46764c4efc9 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 9 Dec 2022 19:33:28 +0000 Subject: [PATCH] debug --- aimodel/src/lib/ai/model_rainfallwater_mono.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py index 8e8ca4b..ad2edc0 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_mono.py +++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py @@ -75,7 +75,7 @@ def model_rainfallwater_mono(metadata, model_arch_enc="convnext_xtiny", model_ar # layer_next = tf.keras.layers.Softmax(axis=-1)(layer_next) # LOSS dice layer_next = tf.keras.layers.Conv2D(1, activation="gelu", kernel_size=1, padding="same")(layer_next) - layer_next = tf.keras.layers.Reshape(tf.constant(layer_next.shape[1:-1]))(layer_next) # Strip the channels + layer_next = tf.keras.layers.Reshape(layer_next.shape[1:-1])(layer_next) # Strip the channels model = tf.keras.Model( inputs = layer_input,