Skip to content

update model imports for new brainio #2183

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

Merged
mike-ferguson merged 3 commits into from
Dec 15, 2025
Merged

update model imports for new brainio #2183

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
9d87dc7
update model imports for new brainio
mike-ferguson Dec 11, 2025
b5b3d25
fix tests
mike-ferguson Dec 11, 2025
4f66289
Merge branch 'master' into s3_rouge_files
mike-ferguson Dec 12, 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
2 changes: 1 addition & 1 deletion brainscore_vision/model_helpers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from brainscore_vision.model_helpers.s3 import load_folder_from_s3, load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_folder_from_s3, load_weight_file

download_weights = load_folder_from_s3
74 changes: 0 additions & 74 deletions brainscore_vision/model_helpers/s3.py
View file
Open in desktop

This file was deleted.

2 changes: 1 addition & 1 deletion brainscore_vision/models/ReAlnet/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
import random
import functools
import torchvision.models
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
import json
Expand Down
4 changes: 2 additions & 2 deletions brainscore_vision/models/ReAlnet_ALL/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,11 @@
from brainscore_vision.model_helpers.check_submission import check_models
import torch.hub
import ssl
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from torch.nn import Module
from .helpers.helpers import TemporalPytorchWrapper
from pathlib import Path
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file
import json
import os

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/ReAlnet_cornet/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from brainscore_vision.model_helpers.check_submission import check_models
import torch.hub
import ssl
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from torch.nn import Module
from .helpers.helpers import TemporalPytorchWrapper
from pathlib import Path
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/Res2Net50_26w_4s/helpers/resnet_helpers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
import torch.nn as nn
import math
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/VOneCORnet_S/helpers/cornet_s_helpers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import torch
from torch.nn import Module
from brainscore_core.supported_data_standards.brainio.assemblies import merge_data_arrays, NeuroidAssembly, walk_coords
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from brainscore_vision.model_helpers.activations.core import ActivationsExtractorHelper
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_01/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_02/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_03/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_04/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_05/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_06/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_07/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_08/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_09/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/alexnet_training_seed_10/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
import numpy as np
from brainscore_vision.model_helpers.s3 import load_weight_file, sha1_hash
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file, sha1_hash
from brainscore_vision.model_helpers.check_submission import check_models
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_cornets_IT/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_cornets_V4/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_resnet50_IT/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_resnet50_V4/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_resnet50_robust_IT/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_resnet50_robust_V4/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_vgg19_IT/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/compact_vgg19_V4/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
import os
import ssl
from brainscore_vision.model_helpers.s3 import load_file
from brainscore_core.supported_data_standards.brainio.s3 import load_file


ssl._create_default_https_context = ssl._create_unverified_context
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/cornet_s/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from brainscore_vision.model_helpers.check_submission import check_models
import torch.hub
import ssl
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from torch.nn import Module
from .helpers.helpers import TemporalPytorchWrapper

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/custom_model_cv_18_dagger_408/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
import torch
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from timm.models import create_model
from brainscore_vision.model_helpers.activations.pytorch import load_images
import ssl
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/effnetb1_cutmix_augmix_sam_e1_5avg_424x377/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
Compose, Normalize, Resize, CenterCrop
)
from albumentations.pytorch import ToTensorV2
from brainscore_vision.model_helpers import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from brainscore_vision.model_helpers.check_submission import check_models

image_resize = 424
Expand Down
2 changes: 1 addition & 1 deletion ..._vision/models/effnetb1_cutmixpatch_SAM_robust32_avge6e8e9e10_manylayers_324x288/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import torch
from brainscore_vision.model_helpers.activations import PytorchWrapper
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
from PIL import Image
import numpy as np
import timm
Expand Down
2 changes: 1 addition & 1 deletion ...re_vision/models/effnetb1_cutmixpatch_augmix_robust32_avge4e7_manylayers_324x288/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
)
from albumentations.pytorch import ToTensorV2
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file


image_resize = 324
Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_4/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet, evnet_params

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_4_no_mapping/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet, evnet_params

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_5/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_6/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_7/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/evresnet_50_8/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.check_submission import check_models
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .evnet.evnet import EVNet

Expand Down
2 changes: 1 addition & 1 deletion brainscore_vision/models/grcnn/model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import functools
from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
from brainscore_vision.model_helpers.s3 import load_weight_file
from brainscore_core.supported_data_standards.brainio.s3 import load_weight_file
import torch
from .helpers.helpers import grcnn55, device

Expand Down
Loading
Loading