Skip to content
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

Add ViG models [NeurIPS 2022] #1578

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

Add ViG models [NeurIPS 2022] #1578

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
cb725a8
add ViG models
iamhankai Dec 5, 2022
f080f1b
update ViG models
iamhankai Dec 6, 2022
1d81cd5
fix error of min input size
iamhankai Dec 6, 2022
38a908b
fix test errors
iamhankai Dec 6, 2022
00101f9
Update vision_gnn.py
iamhankai Dec 7, 2022
33c6d8e
Update gnn_layers.py
iamhankai Dec 7, 2022
df75883
Update gnn_layers.py
iamhankai Dec 7, 2022
7e71058
fix test errors
iamhankai Dec 7, 2022
6c745cc
Merge branch 'main' into main
iamhankai Dec 8, 2022
d26937b
Merge branch 'main' into main
iamhankai Mar 29, 2023
3f46747
add im21k pretrained weights
iamhankai Mar 29, 2023
1daf56d
Update vision_gnn.py
iamhankai Mar 31, 2023
9510681
fix typo
iamhankai Apr 3, 2023
6368a06
Merge branch 'main' into main
iamhankai Apr 3, 2023
0173a13
Merge branch 'main' into main
rwightman Jun 6, 2023
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
8 changes: 1 addition & 7 deletions tests/test_models.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit*',
'poolformer_*', 'volo_*', 'sequencer2d_*', 'pvt_v2*', 'mvitv2*', 'gcvit*', 'efficientformer*',
'eva_*', 'flexivit*', 'eva02*', 'pvig_*'
'eva_*', 'flexivit*', 'eva02*', 'pvig_*', 'samvit_*'
]
NUM_NON_STD = len(NON_STD_FILTERS)

Expand Down Expand Up @@ -295,12 +295,6 @@ def test_model_features_pretrained(model_name, batch_size):
"""Create that pretrained weights load when features_only==True."""
create_model(model_name, pretrained=True, features_only=True)

EXCLUDE_JIT_FILTERS = [
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It seems these lines cannot be removed

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I managed to fix the other jit exceptions so would rather not add more, I feel it's likely it can be supported with appropriate type decl, etc

'*iabn*', 'tresnet*', # models using inplace abn unlikely to ever be scriptable
'dla*', 'hrnet*', 'ghostnet*', 'pvig_*', # hopefully fix at some point
'vit_large_*', 'vit_huge_*', 'vit_gi*',
]


@pytest.mark.torchscript
@pytest.mark.timeout(120)
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.