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

{brainscore_vision-2.2.1/brainscore_vision.egg-info → brainscore_vision-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: brainscore_vision
-Version: 2.2.1
+Version: 2.2.2
 Summary: The Brain-Score library enables model comparisons to behavioral and neural experiments
 License: 
         MIT License

{brainscore_vision-2.2.1 → brainscore_vision-2.2.2}/brainscore_vision/model_helpers/brain_transformation/__init__.py

@@ -79,10 +79,9 @@ class ModelCommitment(BrainModel):
                     self._logger.info(f"Successfully loaded region_layer_map for {identifier}")
                     return json.load(region_layer_map_file)
             else:
-                self._logger.info(f"No region_layer_map file found for {identifier}, proceeding with default layer mapping")
+                self._logger.info(f"No region_layer_map file found for {identifier}. Will proceed with default layer mapping")
                 return None
         except Exception as e:
-            self._logger.error(f"Error importing model to search for region_layer_map: {e}")
             return None
 
     def visual_degrees(self) -> int:

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_0/region_layer_map/yudixie_resnet18_distance_reg_0_240719.json

@@ -0,0 +1 @@
+{"IT": "layer3", "V4": "layer1", "V2": "layer1", "V1": "layer1"}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_1/region_layer_map/yudixie_resnet18_translation_reg_0_240719.json

@@ -0,0 +1 @@
+{"V4": "layer1", "IT": "layer3", "V2": "layer2", "V1": "layer1"}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_10/region_layer_map/yudixie_resnet18_imagenet1kpret_0_240719.json

@@ -0,0 +1 @@
+{"V4": "layer2", "IT": "layer3", "V2": "layer2", "V1": "layer2"}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_11/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_11/region_layer_map/yudixie_resnet18_random_0_240719.json

@@ -0,0 +1,6 @@
+{
+    "V1": "layer1",
+    "V2": "layer2",
+    "V4": "layer1",
+    "IT": "layer4"
+}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_2/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_2/region_layer_map/yudixie_resnet18_rotation_reg_0_240719.json

@@ -0,0 +1 @@
+{"V4": "layer1", "IT": "layer3", "V2": "layer1", "V1": "layer1"}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_2/setup.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [ "torchvision",
+                 "torch"
+]
+
+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.2/brainscore_vision/models/yudixie_resnet18_240719_3/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_3/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_3/region_layer_map/yudixie_resnet18_distance_translation_0_240719.json

@@ -0,0 +1,6 @@
+{
+    "V1": "layer1",
+    "V2": "layer2",
+    "V4": "layer1",
+    "IT": "layer3"
+}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_3/setup.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [ "torchvision",
+                 "torch"
+]
+
+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.2/brainscore_vision/models/yudixie_resnet18_240719_4/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_4/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_4/region_layer_map/yudixie_resnet18_distance_rotation_0_240719.json

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_4/setup.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [ "torchvision",
+                 "torch"
+]
+
+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.2/brainscore_vision/models/yudixie_resnet18_240719_4/test.py

@@ -0,0 +1 @@
+# Left empty as part of 2023 models migration

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_5/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_5/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_5/region_layer_map/yudixie_resnet18_translation_rotation_0_240719.json

@@ -0,0 +1,6 @@
+{
+    "V1": "layer1",
+    "V2": "layer2",
+    "V4": "layer1",
+    "IT": "layer3"
+}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_5/setup.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [ "torchvision",
+                 "torch"
+]
+
+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.2/brainscore_vision/models/yudixie_resnet18_240719_5/test.py

@@ -0,0 +1 @@
+# Left empty as part of 2023 models migration

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_6/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_6/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_6/region_layer_map/yudixie_resnet18_distance_translation_rotation_0_240719.json

@@ -0,0 +1,6 @@
+{
+    "V1": "layer1",
+    "V2": "layer2",
+    "V4": "layer1",
+    "IT": "layer3"
+}

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_6/setup.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+requirements = [ "torchvision",
+                 "torch"
+]
+
+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.2/brainscore_vision/models/yudixie_resnet18_240719_6/test.py

@@ -0,0 +1 @@
+# Left empty as part of 2023 models migration

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_7/__init__.py

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

brainscore_vision-2.2.2/brainscore_vision/models/yudixie_resnet18_240719_7/model.py

@@ -0,0 +1,60 @@
+import os
+from pathlib import Path
+import functools
+from urllib.request import urlretrieve
+
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.models import resnet18
+
+from brainscore_vision.model_helpers.check_submission import check_models
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper, load_preprocess_images
+
+
+# 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):
+    pytorch_device = torch.device('cpu')
+    
+    weigth_url = f'https://yudi-brainscore-models.s3.amazonaws.com/{name}.pth'
+    fh = urlretrieve(weigth_url, f'{name}_weights.pth')
+    load_path = fh[0]
+
+    pytorch_model = resnet18()
+    pytorch_model.fc = nn.Linear(pytorch_model.fc.in_features, 674)
+    pytorch_model = pytorch_model.to(pytorch_device)
+
+    # load model from saved weights
+    saved_state_dict = torch.load(load_path, map_location=pytorch_device)
+    state_dict = {}
+    for k, v in saved_state_dict.items():
+        if k.startswith('_orig_mod.'):
+            # for compiled models
+            state_dict[k[10:]] = v
+        else:
+            state_dict[k] = v
+    pytorch_model.load_state_dict(state_dict, strict=True)
+    print(f'Loaded model from {load_path}')
+
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name,
+                             model=pytorch_model,
+                             preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """xx"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)