From 5dac70aa0871f581d48bbe0b101d35c533f2bca1 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 6 Oct 2022 19:17:03 +0100 Subject: [PATCH] typo --- 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 53826ca..9de247b 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -29,7 +29,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, layer_next = do_convnext_inverse(layer_next, arch_name="convnext_i_tiny") # TODO: An attention layer here instead of a dense layer, with a skip connection perhaps? - logger.warn("Warning: TODO implement attention from https://ieeexplore.ieee.org/document/9076883") + logger.warning("Warning: TODO implement attention from https://ieeexplore.ieee.org/document/9076883") layer_next = tf.keras.layers.Dense(32)(layer_next) layer_next = tf.keras.layers.Conv2D(1, kernel_size=1, activation="softmax", padding="same")(layer_next)