This commit is contained in:
Starbeamrainbowlabs 2022-10-03 17:53:10 +01:00
parent 92c380bff5
commit 2b182214ea
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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)