add vitb14_dinov2_imagenet1k to models

brain-score · Oct 27, 2024 · 2f9a703 · 2f9a703
1 parent 163e211
commit 2f9a703
Show file tree

Hide file tree

Showing 4 changed files with 882 additions and 0 deletions.
diff --git a/brainscore_vision/models/vitb14_dinov2_imagenet1k/__init__.py b/brainscore_vision/models/vitb14_dinov2_imagenet1k/__init__.py
@@ -0,0 +1,5 @@
+from brainscore_vision import model_registry
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from .model import get_model, get_layers
+
+model_registry['vitb14_dinov2_imagenet1k'] = lambda: ModelCommitment(identifier='vitb14_dinov2_imagenet1k', activations_model=get_model('vitb14_dinov2_imagenet1k'), layers=get_layers('vitb14_dinov2_imagenet1k'))