From 92c380bff56c129c44546926382e6a3daf50f189 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 3 Oct 2022 17:51:41 +0100 Subject: [PATCH] fiddle with Conv2DTranspose you need to set the `stride` argument to actually get it to upscale..... :P --- aimodel/src/lib/ai/components/convnext_inverse.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/convnext_inverse.py b/aimodel/src/lib/ai/components/convnext_inverse.py index 296433c..34476d0 100644 --- a/aimodel/src/lib/ai/components/convnext_inverse.py +++ b/aimodel/src/lib/ai/components/convnext_inverse.py @@ -36,7 +36,12 @@ def convnext_inverse(layer_in, depths, dims): def block_upscale(layer_in, block_number, depth, dim): layer_next = layer_in - layer_next = tf.keras.layers.Conv2DTranspose(name=f"cns.stage{block_number}.end.convtp", filters=dim, kernel_size=4, padding="same")(layer_next) + layer_next = tf.keras.layers.Conv2DTranspose( + name=f"cns.stage{block_number}.end.convtp", + filters=dim, + kernel_size=4, + stride=2 + )(layer_next) layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next) for i in range(depth):