From 2b182214eab7bf8ebc7bebc7958e8d835c1919b8 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 3 Oct 2022 17:53:10 +0100 Subject: [PATCH] typo --- aimodel/src/lib/ai/components/convnext_inverse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/convnext_inverse.py b/aimodel/src/lib/ai/components/convnext_inverse.py index 34476d0..b866fb9 100644 --- a/aimodel/src/lib/ai/components/convnext_inverse.py +++ b/aimodel/src/lib/ai/components/convnext_inverse.py @@ -40,7 +40,7 @@ def block_upscale(layer_in, block_number, depth, dim): name=f"cns.stage{block_number}.end.convtp", filters=dim, kernel_size=4, - stride=2 + strides=2 )(layer_next) layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next)