Skip to content

Add SwiftFormer, SHViT, StarNet, FasterNet and GhostNetV3 #2499

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 13 commits into
base: main
Choose a base branch
from
Open

Add SwiftFormer, SHViT, StarNet, FasterNet and GhostNetV3 #2499

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
7a87199
add swiftformer
brianhou0208 Apr 28, 2025
081e6c2
update swiftformer
brianhou0208 Apr 28, 2025
b60a4fb
add starnet
brianhou0208 Apr 28, 2025
77788f4
add shvit
brianhou0208 Apr 28, 2025
91e6e17
add test_models
brianhou0208 May 3, 2025
a3e66b1
add fasternet
brianhou0208 May 5, 2025
7fc0692
update fasternet
brianhou0208 May 5, 2025
89d2952
update group_matcher
brianhou0208 May 14, 2025
cb98094
Merge branch 'main' into fast_model
brianhou0208 May 14, 2025
1922ca5
fix fasternet
brianhou0208 May 14, 2025
236b00c
add ghostnetv3
brianhou0208 May 17, 2025
bea1137
Merge branch 'main' into fast_model
brianhou0208 May 17, 2025
75823ab
Merge branch 'main' into fast_model
brianhou0208 May 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions tests/test_models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,13 +56,13 @@
'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit', 'hieradet_sam2', 'aimv2*', 'tnt',
'tiny_vit', 'vovnet', 'tresnet', 'rexnet', 'resnetv2', 'repghost', 'repvit', 'pvt_v2', 'nextvit', 'nest',
'mambaout', 'inception_next', 'inception_v4', 'hgnet', 'gcvit', 'focalnet', 'efficientformer_v2', 'edgenext',
'davit', 'rdnet', 'convnext', 'pit'
'davit', 'rdnet', 'convnext', 'pit', 'starnet', 'shvit', 'fasternet', 'swiftformer', 'ghostnet',
]

# transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output.
NON_STD_FILTERS = [
'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', 'aimv2*',
'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', 'aimv2*', 'swiftformer_*',
'poolformer_*', 'volo_*', 'sequencer2d_*', 'mvitv2*', 'gcvit*', 'efficientformer*', 'sam_hiera*',
'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*', 'tiny_vit_*', 'hiera_*', 'vitamin*', 'test_vit*',
]
Expand Down Expand Up @@ -221,6 +221,7 @@ def test_model_backward(model_name, batch_size):
EARLY_POOL_MODELS = (
timm.models.EfficientVit,
timm.models.EfficientVitLarge,
timm.models.FasterNet,
timm.models.HighPerfGpuNet,
timm.models.GhostNet,
timm.models.MetaNeXt, # InceptionNeXt
Expand Down
4 changes: 4 additions & 0 deletions timm/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
from .efficientvit_mit import *
from .efficientvit_msra import *
from .eva import *
from .fasternet import *
from .fastvit import *
from .focalnet import *
from .gcvit import *
Expand Down Expand Up @@ -60,7 +61,10 @@
from .selecsls import *
from .senet import *
from .sequencer import *
from .shvit import *
from .sknet import *
from .starnet import *
from .swiftformer import *
from .swin_transformer import *
from .swin_transformer_v2 import *
from .swin_transformer_v2_cr import *
Expand Down
Loading