sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

missing units

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-28 18:17:22 +01:00
parent b5e08f92fe
commit de47a883d9
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/ai/model_rainfallwater_segmentation.py
View file

@ -14,7 +14,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out,
)
# BEGIN
layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense")(layer_input)
layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense", units=feature_dim_in)(layer_input)
layer_next = tf.keras.layers.LayerNormalisation(name="stage_begin.norm", epsilon=1e-6)(layer_next)
layer_next = tf.keras.layers.ReLU(name="stage_begin.relu")(layer_next)