mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 09:13:01 +00:00
typo
This commit is contained in:
parent
92c380bff5
commit
2b182214ea
1 changed files with 1 additions and 1 deletions
|
@ -40,7 +40,7 @@ def block_upscale(layer_in, block_number, depth, dim):
|
||||||
name=f"cns.stage{block_number}.end.convtp",
|
name=f"cns.stage{block_number}.end.convtp",
|
||||||
filters=dim,
|
filters=dim,
|
||||||
kernel_size=4,
|
kernel_size=4,
|
||||||
stride=2
|
strides=2
|
||||||
)(layer_next)
|
)(layer_next)
|
||||||
layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next)
|
layer_next = tf.keras.layers.LayerNormalization(name=f"cns.stage{block_number}.end.norm", epsilon=1e-6)(layer_next)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue