diff --git a/brainscore_vision/models/alexnet/model.py b/brainscore_vision/models/alexnet/model.py index cfa22656b..0941583ea 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='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 index 021deaefb..1e89c812b 100644 --- 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(),