diff --git a/aimodel/src/lib/ai/components/convnext_inverse.py b/aimodel/src/lib/ai/components/convnext_inverse.py index c83fc0c..6f94db8 100644 --- a/aimodel/src/lib/ai/components/convnext_inverse.py +++ b/aimodel/src/lib/ai/components/convnext_inverse.py @@ -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])),