From 0662d0854b0d81869953a4040a5be9fe3d79d804 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 11 Nov 2022 19:11:40 +0000 Subject: [PATCH] model_mono: fix bottleneck --- aimodel/src/lib/ai/model_rainfallwater_mono.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py index 4a6c56e..1ceea01 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_mono.py +++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py @@ -38,10 +38,10 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext arch_name = model_arch_enc )(layer_input) - print("DEBUG:model ENCODER output_shape", layer_next.output_shape) + print("DEBUG:model ENCODER output_shape", layer_next.shape) # BOTTLENECK - layer_next = tf.keras.layers.Dense(name="cns.stage.bottleneck.dense2", units=feature_dim)(layer_input) + layer_next = tf.keras.layers.Dense(name="cns.stage.bottleneck.dense2", units=feature_dim)(layer_next) layer_next = tf.keras.layers.Activation(name="cns.stage.bottleneck.gelu2", activation="gelu")(layer_next) layer_next = tf.keras.layers.LayerNormalization(name="cns.stage.bottleneck.norm2", epsilon=1e-6)(layer_next) layer_next = tf.keras.layers.Dropout(name="cns.stage.bottleneck.dropout", rate=0.1)(layer_next) @@ -50,7 +50,7 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext layer_next = LayerStack2Image(target_width=4, target_height=4)(layer_next) # layer_next = tf.keras.layers.Reshape((4, 4, math.floor(feature_dim_in/(4*4))), name="cns.stable_begin.reshape")(layer_next) - print("DEBUG:model BOTTLENECK:stack2image output_shape", layer_next.output_shape) + print("DEBUG:model BOTTLENECK:stack2image output_shape", layer_next.shape) layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense2", units=feature_dim)(layer_next) layer_next = tf.keras.layers.Activation(name="cns.stage_begin.relu2", activation="gelu")(layer_next)