From 4bb3a70b1295f21440cbdec255c49b4b7c121185 Mon Sep 17 00:00:00 2001 From: Deirdre Kelliher Date: Tue, 18 Jun 2024 13:01:07 -0400 Subject: [PATCH] changing alexnet edit --- brainscore_vision/models/alexnet/__init__.py | 2 +- brainscore_vision/models/alexnet/model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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