Add mobilevit_small.json to region_layer_map for model

brain-score · Oct 14, 2024 · c74f87e · c74f87e
1 parent ba88db6
commit c74f87e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/brainscore_vision/models/mobilevit_small/region_layer_map/mobilevit_small.json b/brainscore_vision/models/mobilevit_small/region_layer_map/mobilevit_small.json
@@ -0,0 +1,6 @@
+{
+    "V1": "mobilevit.encoder.layer.3",
+    "V2": "mobilevit.encoder.layer.4",
+    "V4": "mobilevit.encoder.layer.3",
+    "IT": "mobilevit.encoder.layer.4"
+}