Layer Mapping Orchestrator (build:114) (#1399)

* Add region_layer_map directories with .gitkeep for models: resnet18_imagenet_full resnet34_imagenet_full resnet50_imagenet_full * Add resnet18_imagenet_full.json to region_layer_map for model resnet18_imagenet_full * Add resnet34_imagenet_full.json to region_layer_map for model resnet34_imagenet_full * Add resnet50_imagenet_full.json to region_layer_map for model resnet50_imagenet_full * Delete brainscore_vision/models/scaling_models/region_layer_map/.gitkeep --------- Co-authored-by: Jenkins <[email protected]> Co-authored-by: Kartik Pradeepan <[email protected]>
brain-score · Oct 29, 2024 · 1ed5a5c · 1ed5a5c
1 parent 6323907
commit 1ed5a5c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/brainscore_vision/models/scaling_models/region_layer_map/resnet18_imagenet_full.json b/brainscore_vision/models/scaling_models/region_layer_map/resnet18_imagenet_full.json
@@ -0,0 +1,6 @@
+{
+    "V1": "layer1.0.bn1",
+    "V2": "layer2.0.bn2",
+    "V4": "layer3.0.conv2",
+    "IT": "layer4.0.bn1"
+}
diff --git a/brainscore_vision/models/scaling_models/region_layer_map/resnet34_imagenet_full.json b/brainscore_vision/models/scaling_models/region_layer_map/resnet34_imagenet_full.json
@@ -0,0 +1,6 @@
+{
+    "V1": "layer1.0.bn1",
+    "V2": "layer3.1.conv1",
+    "V4": "layer3.1.conv1",
+    "IT": "layer4.0.bn1"
+}
diff --git a/brainscore_vision/models/scaling_models/region_layer_map/resnet50_imagenet_full.json b/brainscore_vision/models/scaling_models/region_layer_map/resnet50_imagenet_full.json
@@ -0,0 +1,6 @@
+{
+    "V1": "layer1.0.conv1",
+    "V2": "layer3.5.bn3",
+    "V4": "layer3.0.conv1",
+    "IT": "layer4.0.relu"
+}