sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-07-02 11:24:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

segmentation: adds xxtiny, but unsure if it's small enough

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-11 19:22:37 +01:00
parent f4a2c742d9
commit c45b90764e
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -6,6 +6,7 @@ from .convnext import add_convnext_block
depths_dims = dict(
# architectures from: https://github.com/facebookresearch/ConvNeXt
# A ConvNet for the 2020s: https://arxiv.org/abs/2201.03545
convnext_i_xxtiny = (dict(depths=[3, 6, 3, 3], dims=[264, 132, 96, 66])),
convnext_i_xtiny = (dict(depths=[3, 6, 3, 3], dims=[528, 264, 132, 66])),
convnext_i_tiny = (dict(depths=[3, 9, 3, 3], dims=[768, 384, 192, 96])),
convnext_i_small = (dict(depths=[3, 27, 3, 3], dims=[768, 384, 192, 96])),