From af77cc3c83f6f9f9f29352becb354a26b65d7298 Mon Sep 17 00:00:00 2001 From: Katherine Fairchild Date: Thu, 9 May 2024 07:05:30 -0400 Subject: [PATCH] add resnet18_imagenet21kP to models (#818) Co-authored-by: AutoJenkins Co-authored-by: Sam Winebrake <85908068+samwinebrake@users.noreply.github.com> --- brainscore_vision/models/resnet18_imagenet21kP/__init__.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/brainscore_vision/models/resnet18_imagenet21kP/__init__.py b/brainscore_vision/models/resnet18_imagenet21kP/__init__.py index be444ccee..4b3207d97 100644 --- 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["resnet18_imagenet21kP-abdulkadir.gokce@epfl.ch"] = lambda: get_model( - MODEL_NAME +model_registry["resnet18_imagenet21kP"] = lambda: get_model( + "resnet18_imagenet21kP" )