add resnet18_imagenet21kP to models (#818)

Co-authored-by: AutoJenkins <[email protected]> Co-authored-by: Sam Winebrake <[email protected]>
brain-score · May 9, 2024 · af77cc3 · af77cc3
1 parent de683f5
commit af77cc3
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/brainscore_vision/models/resnet18_imagenet21kP/__init__.py b/brainscore_vision/models/resnet18_imagenet21kP/__init__.py
@@ -1,8 +1,6 @@
 from brainscore_vision import model_registry
 from .model import get_model
 
-MODEL_NAME = "resnet18_imagenet21kP"
-
-model_registry["[email protected]"] = lambda: get_model(
-    MODEL_NAME
+model_registry["resnet18_imagenet21kP"] = lambda: get_model(
+    "resnet18_imagenet21kP"
 )