From 527b34942d4764544215bfcaa7ccae506274a0c8 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 11 Nov 2022 19:29:37 +0000 Subject: [PATCH] =?UTF-8?q?convnext=5Finverse:=20kernel=5Fsize=204?= =?UTF-8?q?=E2=86=922?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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 6f94db8..b67d697 100644 --- a/aimodel/src/lib/ai/components/convnext_inverse.py +++ b/aimodel/src/lib/ai/components/convnext_inverse.py @@ -41,7 +41,7 @@ def block_upscale(layer_in, block_number, depth, dim): layer_next = tf.keras.layers.Conv2DTranspose( name=f"cns.stage{block_number}.end.convtp", filters=dim, - kernel_size=4, + kernel_size=2, # double each time, was 4 strides=2 )(layer_next) layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next)