Merge branch 'master' into coggan_behavior_refactor

.github/workflows/score_new_plugins.yml

@@ -91,7 +91,7 @@ jobs:
           echo "$PLUGIN_INFO" > plugin-info.json
       
       - name: Upload PLUGIN_INFO as an artifact
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           name: plugin-info
           path: plugin-info.json
@@ -114,7 +114,7 @@ jobs:
     needs: extract_email
     steps:
       - name: Download PLUGIN_INFO artifact
-        uses: actions/download-artifact@v2
+        uses: actions/download-artifact@v4
         with:
           name: plugin-info
           path: artifact-directory
@@ -144,7 +144,7 @@ jobs:
           echo "$PLUGIN_INFO" > plugin-info.json
       
       - name: Upload PLUGIN_INFO as an artifact
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           name: plugin-info
           path: plugin-info.json
@@ -161,7 +161,7 @@ jobs:
     steps:
 
       - name: Download PLUGIN_INFO artifact
-        uses: actions/download-artifact@v2
+        uses: actions/download-artifact@v4
         with:
           name: plugin-info
           path: artifact-directory

brainscore_vision/models/resnet50_11ad3fa6/__init__.py

@@ -0,0 +1,9 @@
+from brainscore_vision import model_registry
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from .model import get_model, get_layers
+
+model_registry['resnet50_11ad3fa6'] = lambda: ModelCommitment(
+    identifier='resnet50_11ad3fa6',
+    activations_model=get_model('resnet50_11ad3fa6'),
+    layers=get_layers('resnet50_11ad3fa6')
+)

brainscore_vision/models/resnet50_11ad3fa6/model.py

@@ -0,0 +1,35 @@
+from brainscore_vision.model_helpers.check_submission import check_models
+import functools
+import torchvision.models
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
+
+# This is an example implementation for submitting resnet-50 as a pytorch model
+
+# Attention: It is important, that the wrapper identifier is unique per model!
+# The results will otherwise be the same due to brain-scores internal result caching mechanism.
+# Please load your pytorch model for usage in CPU. There won't be GPUs available for scoring your model.
+# If the model requires a GPU, contact the brain-score team directly.
+
+
+def get_model(name):
+    assert name == 'resnet50_11ad3fa6'
+    model = torchvision.models.resnet50(pretrained=True)
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(
+        identifier='resnet50_11ad3fa6', model=model, preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    assert name == 'resnet50_11ad3fa6'
+    return ['conv1', 'layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision/models/resnet50_11ad3fa6/requirements.txt

@@ -0,0 +1,2 @@
+torchvision
+torch

brainscore_vision/models/resnet50_11ad3fa6/test.py

@@ -0,0 +1,8 @@
+import pytest
+import brainscore_vision
+
+
+@pytest.mark.travis_slow
+def test_has_identifier():
+    model = brainscore_vision.load_model('resnet50_tutorial')
+    assert model.identifier == 'resnet50_tutorial'

environment_lock.yml

@@ -34,9 +34,9 @@ dependencies:
       - bleach==6.1.0
       - boto3==1.35.3
       - botocore==1.35.3
-      - brainio @ git+https://github.com/brain-score/brainio.git@main
-      - brainscore_core @ git+https://github.com/brain-score/core@main
-      - brainscore-vision @ git+https://github.com/brain-score/vision.git@main
+      - brainscore-brainio==1.0.0
+      - brainscore-core==2.1
+      - brainscore-vision==2.1
       - certifi==2024.7.4
       - cffi==1.17.0
       - cftime==1.6.4
@@ -144,7 +144,7 @@ dependencies:
       - qtpy==2.4.1
       - referencing==0.35.1
       - requests==2.32.3
-      - result_caching @ git+https://github.com/brain-score/result_caching@master
+      - result_caching==0.1.0
       - rfc3339-validator==0.1.4
       - rfc3986-validator==0.1.1
       - rpds-py==0.20.0