From 2a1772a21187fe027440cba90c4109dec40efbfb Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 8 Dec 2022 19:10:12 +0000 Subject: [PATCH] confvnext_intrevse: add shallow --- aimodel/src/lib/ai/components/convnext_inverse.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/aimodel/src/lib/ai/components/convnext_inverse.py b/aimodel/src/lib/ai/components/convnext_inverse.py index b67d697..c6e3df0 100644 --- a/aimodel/src/lib/ai/components/convnext_inverse.py +++ b/aimodel/src/lib/ai/components/convnext_inverse.py @@ -6,13 +6,14 @@ 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])), - convnext_i_base = (dict(depths=[3, 27, 3, 3], dims=[1024, 512, 256, 128])), - convnext_i_large = (dict(depths=[3, 27, 3, 3], dims=[1536, 768, 384, 192])), - convnext_i_xlarge = (dict(depths=[3, 27, 3, 3], dims=[2048, 1024, 512, 256])), + convnext_i_shallow= (dict(depths=[2, 3, 2], dims=[132, 96, 66])), + 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])), + convnext_i_base = (dict(depths=[3, 27, 3, 3], dims=[1024, 512, 256, 128])), + convnext_i_large = (dict(depths=[3, 27, 3, 3], dims=[1536, 768, 384, 192])), + convnext_i_xlarge = (dict(depths=[3, 27, 3, 3], dims=[2048, 1024, 512, 256])), )