mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Rename inception_next_atto pretrained str

Browse Source
This commit is contained in:
Ross Wightman 2024-12-06 10:08:03 -08:00 committed by Ross Wightman
parent 0576175d85
commit d1e9a8622a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/inception_next.py
View File

@ -415,7 +415,7 @@ def inception_next_atto(pretrained=False, **kwargs):
depths=(2, 2, 6, 2), dims=(40, 80, 160, 320), depths=(2, 2, 6, 2), dims=(40, 80, 160, 320),
token_mixers=partial(InceptionDWConv2d, band_kernel_size=9, branch_ratio=0.25) 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 @register_model