From d7a7ed7ba95fb477dcfd4b20743e847e1990112a Mon Sep 17 00:00:00 2001 From: Weihao Yu Date: Fri, 6 Dec 2024 14:22:29 +0800 Subject: [PATCH 1/2] Add inception_next_atto --- timm/models/inception_next.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/timm/models/inception_next.py b/timm/models/inception_next.py index ec5d49e987..6b2f045b2e 100644 --- a/timm/models/inception_next.py +++ b/timm/models/inception_next.py @@ -375,6 +375,10 @@ def _cfg(url='', **kwargs): default_cfgs = generate_default_cfgs({ + 'inception_next_atto.sail_in1k': _cfg( + hf_hub_id='timm/', + # url='https://github.com/sail-sg/inceptionnext/releases/download/model/inceptionnext_atto.pth', + ), 'inception_next_tiny.sail_in1k': _cfg( hf_hub_id='timm/', # url='https://github.com/sail-sg/inceptionnext/releases/download/model/inceptionnext_tiny.pth', @@ -405,6 +409,15 @@ def _create_inception_next(variant, pretrained=False, **kwargs): return model +@register_model +def inception_next_atto(pretrained=False, **kwargs): + model_args = dict( + depths=(2, 2, 6, 2), dims=(40, 80, 160, 320), + token_mixers=partial(InceptionDWConv2d, band_kernel_size=9, branch_ratio=0.25) + ) + return _create_inception_next('inception_next_tiny', pretrained=pretrained, **dict(model_args, **kwargs)) + + @register_model def inception_next_tiny(pretrained=False, **kwargs): model_args = dict( From e90b68b60397b16efabd3ea396b25e2525adf9a5 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 6 Dec 2024 10:08:03 -0800 Subject: [PATCH 2/2] Rename inception_next_atto pretrained str --- timm/models/inception_next.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/inception_next.py b/timm/models/inception_next.py index 6b2f045b2e..3c4906aa05 100644 --- a/timm/models/inception_next.py +++ b/timm/models/inception_next.py @@ -415,7 +415,7 @@ def inception_next_atto(pretrained=False, **kwargs): depths=(2, 2, 6, 2), dims=(40, 80, 160, 320), token_mixers=partial(InceptionDWConv2d, band_kernel_size=9, branch_ratio=0.25) ) - return _create_inception_next('inception_next_tiny', pretrained=pretrained, **dict(model_args, **kwargs)) + return _create_inception_next('inception_next_atto', pretrained=pretrained, **dict(model_args, **kwargs)) @register_model