From a336cdee9015711a23745ec8262a4624f17ee593 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 18:18:10 +0100 Subject: [PATCH] and continues --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index 3f0a555..92ae012 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -15,7 +15,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, # BEGIN layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense", units=feature_dim_in)(layer_input) - layer_next = tf.keras.layers.LayerNormalisation(name="stage_begin.norm", epsilon=1e-6)(layer_next) + layer_next = tf.keras.layers.LayerNormalization(name="stage_begin.norm", epsilon=1e-6)(layer_next) layer_next = tf.keras.layers.ReLU(name="stage_begin.relu")(layer_next) layer_next = do_convnext_inverse(layer_next, arch_name="convnext_i_tiny")