From 85012d06168b53da9dbc14010e90b79a581e2b58 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 9 Dec 2022 19:18:03 +0000 Subject: [PATCH] fixup --- 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 bc7365e..52c619e 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_input = tf.keras.layers.Reshape(layer_next.shape[:-1]) # Strip the channels + layer_next = tf.keras.layers.Reshape(layer_next.shape[:-1])(layer_next) # Strip the channels model = tf.keras.Model( inputs = layer_input,