changing alexnet edit

brain-score · Jun 18, 2024 · 4bb3a70 · 4bb3a70
1 parent 575a139
commit 4bb3a70
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/brainscore_vision/models/alexnet/__init__.py b/brainscore_vision/models/alexnet/__init__.py
@@ -2,7 +2,7 @@
 from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
 from .model import get_model, LAYERS
 
-model_registry['alexnet'] = lambda: ModelCommitment(
+model_registry['alexne'] = lambda: ModelCommitment(
     identifier='alexnet',
     activations_model=get_model(),
     layers=LAYERS)
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='alexne', model=model, preprocessing=preprocessing)
+    wrapper = PytorchWrapper(identifier='alexnet', model=model, preprocessing=preprocessing)
     wrapper.image_size = 224
     return wrapper