diff --git a/brainscore_vision/models/alexnet/__init__.py b/brainscore_vision/models/alexnet/__init__.py index e8a0173fc..150f14ba4 100644 --- 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 index 0941583ea..cfa22656b 100644 --- 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