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

model_contrastive: make water encoder significantly shallower

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-24 20:52:31 +01:00
parent a6b07a49cb
commit f2e2ca1484
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
aimodel/src/lib/ai/components/convnext.py
View file

@ -11,6 +11,7 @@ kernel_initial = tf.keras.initializers.TruncatedNormal(stddev=0.2)
bias_initial = tf.keras.initializers.Constant(value=0)
depths_dims = dict(
convnext_shallow= (dict(depths=[1, 3, 1], dims=[66, 132, 192])),
convnext_xtiny = (dict(depths=[3, 3, 6, 3], dims=[66, 132, 264, 528])),
# architectures from: https://github.com/facebookresearch/ConvNeXt
# A ConvNet for the 2020s: https://arxiv.org/abs/2201.03545

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

@ -44,7 +44,7 @@ def model_rainfallwater_contrastive(metadata, shape_water, batch_size=64, featur
input_height=water_height,
input_channels=water_channels,
feature_dim=feature_dim,
arch_name="convnext_xtiny",
arch_name="convnext_shallow",
summary_file=summary_file
)(input_water)