adding two models to be tested

brain-score · Jun 12, 2024 · 3bd9b38 · 3bd9b38
1 parent 58232eb
commit 3bd9b38
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/brainscore_vision/models/alexnet/model.py b/brainscore_vision/models/alexnet/model.py
@@ -23,6 +23,6 @@
 def get_model():
     model = torchvision.models.alexnet(pretrained=True)
     preprocessing = functools.partial(load_preprocess_images, image_size=224)
-    wrapper = PytorchWrapper(identifier='alexnet', model=model, preprocessing=preprocessing)
+    wrapper = PytorchWrapper(identifier='alexne', model=model, preprocessing=preprocessing)
     wrapper.image_size = 224
     return wrapper
diff --git a/brainscore_vision/models/alexnet_ks_torevert/__init__.py b/brainscore_vision/models/alexnet_ks_torevert/__init__.py
@@ -2,6 +2,7 @@
 from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
 from .model import get_model, LAYERS
 
+
 model_registry['alexnet_ks_torevert'] = lambda: ModelCommitment(
     identifier='alexnet',
     activations_model=get_model(),