brainscore-vision 2.2.2__tar.gz → 2.2.3__tar.gz

brainscore_vision-2.2.3/PKG-INFO

@@ -0,0 +1,153 @@
+Metadata-Version: 2.1
+Name: brainscore_vision
+Version: 2.2.3
+Summary: The Brain-Score library enables model comparisons to behavioral and neural experiments
+License: 
+        MIT License
+        
+        Copyright (c) 2017, Martin Schrimpf and Jon Prescott-Roy
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+        
+        
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy<2
+Requires-Dist: brainscore-core
+Requires-Dist: result-caching
+Requires-Dist: importlib-metadata<5
+Requires-Dist: scikit-learn
+Requires-Dist: scipy
+Requires-Dist: opencv-python
+Requires-Dist: h5py
+Requires-Dist: tqdm
+Requires-Dist: gitpython
+Requires-Dist: fire
+Requires-Dist: jupyter
+Requires-Dist: pybtex
+Requires-Dist: peewee
+Requires-Dist: psycopg2-binary
+Requires-Dist: networkx
+Requires-Dist: eva-decord
+Requires-Dist: psutil
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest_check; extra == "test"
+Requires-Dist: pytest-mock; extra == "test"
+Requires-Dist: pytest-timeout; extra == "test"
+Requires-Dist: torch; extra == "test"
+Requires-Dist: torchvision; extra == "test"
+Requires-Dist: matplotlib; extra == "test"
+Requires-Dist: pytest-mock; extra == "test"
+
+[![Build Status](https://app.travis-ci.com/brain-score/vision.svg?token=vqt7d2yhhpLGwHsiTZvT&branch=master)](https://app.travis-ci.com/brain-score/vision)
+[![Documentation Status](https://readthedocs.org/projects/brain-score/badge/?version=latest)](https://brain-score.readthedocs.io/en/latest/?badge=latest)
+[![Contributor Covenant](https://img.shields.io/badge/Contributor%20Covenant-2.1-4baaaa.svg)](code_of_conduct.md) 
+
+Brain-Score is a platform to evaluate computational models of brain function 
+on their match to brain measurements in primate vision. 
+The intent of Brain-Score is to adopt many (ideally all) the experimental benchmarks in the field
+for the purpose of model testing, falsification, and comparison.
+To that end, Brain-Score operationalizes experimental data into quantitative benchmarks 
+that any model candidate following the [`BrainModel`](brainscore_vision/model_interface.py) interface can be scored on.
+
+Note that you can only access a limited set of public benchmarks when running locally. To score a model on all benchmarks, submit it via the [brain-score.org website](http://www.brain-score.org).
+
+See the [documentation](https://brain-score.readthedocs.io) for more details, e.g. for submitting a [model](https://brain-score.readthedocs.io/en/latest/modules/model_tutorial.html) or [benchmark](https://brain-score.readthedocs.io/en/latest/modules/benchmark_tutorial.html) to Brain-Score. For a step-by-step walkthrough on submitting models to the Brain-Score website, see these [web tutorials](https://www.brain-score.org/tutorial/).
+
+See these [code examples](https://github.com/brain-score/vision/blob/master/examples) on scoring models, retrieving data, using and defining benchmarks and metrics.
+These [previous examples](https://github.com/brain-score/candidate_models/blob/master/examples/score-model.ipynb) might be helpful, but their usage has been deprecated after the 2.0 update.
+
+Brain-Score is made by and for the community. 
+To contribute, please [send in a pull request](https://github.com/brain-score/vision/pulls).
+
+
+## Local installation
+
+You will need Python = 3.11 and pip >= 18.1.
+
+`pip install git+https://github.com/brain-score/vision.git`
+
+Test if the installation is successful by scoring a model on a public benchmark:
+
+```python
+from brainscore_vision.benchmarks import public_benchmark_pool
+
+benchmark = public_benchmark_pool['dicarlo.MajajHong2015public.IT-pls']
+model = my_model()
+score = benchmark(model)
+
+# >  <xarray.Score ()>
+# >  array(0.07637264)
+# >  Attributes:
+# >  Attributes:
+# >      error:                 <xarray.Score ()>\narray(0.00548197)
+# >      raw:                   <xarray.Score ()>\narray(0.22545106)\nAttributes:\...
+# >      ceiling:               <xarray.DataArray ()>\narray(0.81579938)\nAttribut...
+# >      model_identifier:      my-model
+# >      benchmark_identifier:  dicarlo.MajajHong2015public.IT-pls
+```
+
+Some steps may take minutes because data has to be downloaded during first-time use.
+
+## Environment Variables
+
+| Variable               | Description                                                                                                                           |
+|------------------------|---------------------------------------------------------------------------------------------------------------------------------------|
+| RESULTCACHING_HOME     | directory to cache results (benchmark ceilings) in, `~/.result_caching` by default (see https://github.com/brain-score/result_caching) |
+
+
+## License
+
+MIT license
+
+
+## Troubleshooting
+
+<details>
+<summary>`ValueError: did not find HDF5 headers` during netcdf4 installation</summary>
+pip seems to fail properly setting up the HDF5_DIR required by netcdf4.
+Use conda: `conda install netcdf4`
+</details>
+
+<details>
+<summary>repeated runs of a benchmark / model do not change the outcome even though code was changed</summary>
+results (scores, activations) are cached on disk using https://github.com/mschrimpf/result_caching.
+Delete the corresponding file or directory to clear the cache.
+</details>
+
+
+## CI environment
+
+Add CI related build commands to `test_setup.sh`. The script is executed in CI environment for unittests.
+
+
+## References
+
+If you use Brain-Score in your work, please cite 
+["Brain-Score: Which Artificial Neural Network for Object Recognition is most Brain-Like?"](https://www.biorxiv.org/content/10.1101/407007v2) (technical) and 
+["Integrative Benchmarking to Advance Neurally Mechanistic Models of Human Intelligence"](https://www.cell.com/neuron/fulltext/S0896-6273(20)30605-X) (perspective) 
+as well as the respective benchmark sources.
+
+```bibtex
+@article{SchrimpfKubilius2018BrainScore,
+  title={Brain-Score: Which Artificial Neural Network for Object Recognition is most Brain-Like?},
+  author={Martin Schrimpf and Jonas Kubilius and Ha Hong and Najib J. Majaj and Rishi Rajalingham and Elias B. Issa and Kohitij Kar and Pouya Bashivan and Jonathan Prescott-Roy and Franziska Geiger and Kailyn Schmidt and Daniel L. K. Yamins and James J. DiCarlo},
+  journal={bioRxiv preprint},
+  year={2018},
+  url={https://www.biorxiv.org/content/10.1101/407007v2}
+}
+
+@article{Schrimpf2020integrative,
+  title={Integrative Benchmarking to Advance Neurally Mechanistic Models of Human Intelligence},
+  author={Schrimpf, Martin and Kubilius, Jonas and Lee, Michael J and Murty, N Apurva Ratan and Ajemian, Robert and DiCarlo, James J},
+  journal={Neuron},
+  year={2020},
+  url={https://www.cell.com/neuron/fulltext/S0896-6273(20)30605-X}
+}
+```

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_1/__init__.py

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

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_1/model.py

@@ -0,0 +1,200 @@
+
+from brainscore_vision.model_helpers.check_submission import check_models
+import functools
+import numpy as np
+import torch
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+from PIL import Image
+from torch import nn
+import pytorch_lightning as pl
+import torchvision.models as models
+import gdown
+import glob
+import os
+from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
+
+def get_bibtex(model_identifier):
+    return 'VGG16'
+
+def get_model_list():
+    return ['alexnet_less_variation_iteration=1']
+
+def get_model(name):
+    keyword = 'less_variation'
+    iteration = 1
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_1.ckpt'
+    output = 'alexnet_less_variation_iteration=1.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=1.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        layers = []
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    assert name == 'alexnet_less_variation_iteration=1'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_1.ckpt'
+    output = 'alexnet_less_variation_iteration=1.ckpt'
+    gdown.download(url, output)
+    layers = []
+    for name, module in model._modules.items():
+        print(name, "->", module)
+        layers.append(name)
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    activations_model = PytorchWrapper(identifier=name, model=model, preprocessing=preprocessing)
+
+    return activations_model
+
+def get_layers(name):
+    keyword = 'less_variation'
+    iteration = 1
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_1.ckpt'
+    output = 'alexnet_less_variation_iteration=1.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=1.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    layers = []
+    for name, module in model._modules.items():
+            print(name, "->", module)
+            layers.append(name)
+    return layers
+
+if __name__ == '__main__':
+    device = "cpu"
+    global model
+    global keyword
+    global network
+    global iteration
+    keyword = 'less_variation'
+    iteration = 1
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_1.ckpt'
+    output = 'alexnet_less_variation_iteration=1.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=1.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        layers = []
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_1/region_layer_map/alexnet_less_variation_iteration=1.json

@@ -0,0 +1,6 @@
+{
+    "V1": "features",
+    "V2": "features",
+    "V4": "features",
+    "IT": "features"
+}

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_1/setup.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [
+    "torchvision",
+    "torch",
+    "gdown",
+    "pytorch_lightning",
+    "brainscore_vision"
+]
+
+setup(
+    packages=find_packages(exclude=['tests']),
+    include_package_data=True,
+    install_requires=requirements,
+    license="MIT license",
+    zip_safe=False,
+    keywords='brain-score template',
+    classifiers=[
+        'Development Status :: 2 - Pre-Alpha',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: MIT License',
+        'Natural Language :: English',
+        'Programming Language :: Python :: 3.7',
+    ],
+    test_suite='tests',
+)

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_1/test.py

@@ -0,0 +1,3 @@
+
+import pytest
+

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_2/__init__.py

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

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_2/model.py

@@ -0,0 +1,200 @@
+
+from brainscore_vision.model_helpers.check_submission import check_models
+import functools
+import numpy as np
+import torch
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+from PIL import Image
+from torch import nn
+import pytorch_lightning as pl
+import torchvision.models as models
+import gdown
+import glob
+import os
+from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
+
+def get_bibtex(model_identifier):
+    return 'VGG16'
+
+def get_model_list():
+    return ['alexnet_less_variation_iteration=2']
+
+def get_model(name):
+    keyword = 'less_variation'
+    iteration = 2
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_2.ckpt'
+    output = 'alexnet_less_variation_iteration=2.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=2.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        layers = []
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    assert name == 'alexnet_less_variation_iteration=2'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_2.ckpt'
+    output = 'alexnet_less_variation_iteration=2.ckpt'
+    gdown.download(url, output)
+    layers = []
+    for name, module in model._modules.items():
+        print(name, "->", module)
+        layers.append(name)
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    activations_model = PytorchWrapper(identifier=name, model=model, preprocessing=preprocessing)
+
+    return activations_model
+
+def get_layers(name):
+    keyword = 'less_variation'
+    iteration = 2
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_2.ckpt'
+    output = 'alexnet_less_variation_iteration=2.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=2.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    layers = []
+    for name, module in model._modules.items():
+            print(name, "->", module)
+            layers.append(name)
+    return layers
+
+if __name__ == '__main__':
+    device = "cpu"
+    global model
+    global keyword
+    global network
+    global iteration
+    keyword = 'less_variation'
+    iteration = 2
+    network = 'alexnet'
+    url = 'https://eggerbernhard.ch/shreya/latest_alexnet/less_variation_2.ckpt'
+    output = 'alexnet_less_variation_iteration=2.ckpt'
+    gdown.download(url, output)
+
+
+    if keyword != 'imagenet_trained' and keyword != 'no_training':
+        lx_whole = [f"alexnet_less_variation_iteration=2.ckpt"]
+        if len(lx_whole) > 1:
+            lx_whole = [lx_whole[-1]]
+    elif keyword == 'imagenet_trained' or keyword == 'no_training':
+        print('keyword is imagenet')
+        lx_whole = ['x']
+
+    for model_ckpt in lx_whole:
+        print(model_ckpt)
+        last_module_name = None
+        last_module = None
+        layers = []
+        if keyword == 'imagenet_trained' and network != 'clip':
+            model = torch.hub.load('pytorch/vision', network, pretrained=True)
+            for name, module in model.named_modules():
+                last_module_name = name
+                last_module = module
+                layers.append(name)
+        else:
+            model = torch.hub.load('pytorch/vision', network, pretrained=False)
+        if model_ckpt != 'x':
+            ckpt = torch.load(model_ckpt, map_location='cpu')
+        if model_ckpt != 'x' and network == 'alexnet' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        if model_ckpt != 'x' and network == 'vgg16' and keyword != 'imagenet_trained':
+            ckpt2 = {}
+            for keys in ckpt['state_dict']:
+                print(keys)
+                print(ckpt['state_dict'][keys].shape)
+                print('---')
+                k2 = keys.split('model.')[1]
+                ckpt2[k2] = ckpt['state_dict'][keys]
+            model.load_state_dict(ckpt2)
+        # Add more cases for other networks as needed
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_2/region_layer_map/alexnet_less_variation_iteration=2.json

@@ -0,0 +1,6 @@
+{
+    "V1": "features",
+    "V2": "features",
+    "V4": "features",
+    "IT": "features"
+}

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_2/setup.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [
+    "torchvision",
+    "torch",
+    "gdown",
+    "pytorch_lightning",
+    "brainscore_vision"
+]
+
+setup(
+    packages=find_packages(exclude=['tests']),
+    include_package_data=True,
+    install_requires=requirements,
+    license="MIT license",
+    zip_safe=False,
+    keywords='brain-score template',
+    classifiers=[
+        'Development Status :: 2 - Pre-Alpha',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: MIT License',
+        'Natural Language :: English',
+        'Programming Language :: Python :: 3.7',
+    ],
+    test_suite='tests',
+)

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_2/test.py

@@ -0,0 +1,3 @@
+
+import pytest
+

brainscore_vision-2.2.3/brainscore_vision/models/alexnet_less_variation_4/__init__.py

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