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

Custom trained models #1206

Merged
merged 29 commits into from
Sep 17, 2024
+1,282 −0
Merged

Custom trained models #1206

Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
a84d7d3
Custom trained models
akgokce Aug 29, 2024
04187f0
Fix missing library
akgokce Aug 29, 2024
eee6ad2
Fix a missing library
akgokce Aug 29, 2024
47cae96
Not every model needs timm
akgokce Aug 29, 2024
fe7aadd
Trigger build tests
akgokce Aug 29, 2024
52d50e9
Use model_id instead of model_name to ensure correct weights are loaded
akgokce Aug 29, 2024
4a2d151
timm is missing from deit models
akgokce Aug 29, 2024
1777ac5
timm is missing from deit models
akgokce Aug 29, 2024
3e5d152
Fix requirements.txt newline
akgokce Aug 29, 2024
089c224
Merge branch 'master' into master
akgokce Aug 29, 2024
62f81f7
Package models into a single plugin
akgokce Sep 9, 2024
5474f00
Merge branch 'master' of https://github.com/akgokce/vision-custom-models
akgokce Sep 9, 2024
9cc1cd1
Merge branch 'brain-score:master' into master
akgokce Sep 9, 2024
980d740
Update requirements.txt
akgokce Sep 10, 2024
75bcdab
Merge branch 'master' of https://github.com/akgokce/vision-custom-models
akgokce Sep 10, 2024
0158d9d
Update requirements.txt
akgokce Sep 10, 2024
466f297
Merge branch 'master' into master
akgokce Sep 10, 2024
9701fb5
Update requirements.txt
akgokce Sep 10, 2024
f28069d
Change the order of models
akgokce Sep 10, 2024
dd6346d
Merge branch 'master' into master
akgokce Sep 11, 2024
d3b71d5
Update model.py
akgokce Sep 11, 2024
a5b62af
Merge branch 'master' of https://github.com/akgokce/vision-custom-models
akgokce Sep 11, 2024
7910891
Disable resultcaching for vits
akgokce Sep 12, 2024
041ba09
Merge branch 'master' into master
akgokce Sep 12, 2024
519d83d
Tests reordering
akgokce Sep 12, 2024
e044547
Revert tests back
akgokce Sep 15, 2024
16ff06f
Merge branch 'master' into master
akgokce Sep 15, 2024
9051860
Update model.py
akgokce Sep 15, 2024
414ad2b
Update model.py
akgokce Sep 15, 2024
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
Prev Previous commit
Next Next commit
Loading
Loading status checks…
Fix requirements.txt newline
@akgokce
akgokce committed Aug 29, 2024
commit 3e5d1521357001da588e96f85793bdeb24761be0
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_base_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_large_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_small_imagenet_100_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_small_imagenet_10_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_small_imagenet_1_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_small_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/convnext_tiny_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_base_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_large_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_small_imagenet_100_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_small_imagenet_10_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_small_imagenet_1_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/deit_small_imagenet_full_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
torch
torchvision
albumentations
timm
timm
2 changes: 1 addition & 1 deletion brainscore_vision/models/efficientnet_b0_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/efficientnet_b1_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/efficientnet_b2_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet101_ecoset_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet101_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet152_ecoset_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet152_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet18_ecoset_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet18_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet34_ecoset_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet34_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet50_ecoset_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet50_imagenet_100_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet50_imagenet_10_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet50_imagenet_1_seed-0/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations
2 changes: 1 addition & 1 deletion brainscore_vision/models/resnet50_imagenet_full/requirements.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
torch
torchvision
albumentations
albumentations