From 1ed5a5c7e2a9b8daf405c513f4f6d6c0ccb5087e Mon Sep 17 00:00:00 2001 From: Katherine Fairchild Date: Tue, 29 Oct 2024 05:06:31 -0400 Subject: [PATCH] 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 Co-authored-by: Kartik Pradeepan --- .../region_layer_map/resnet18_imagenet_full.json | 6 ++++++ .../region_layer_map/resnet34_imagenet_full.json | 6 ++++++ .../region_layer_map/resnet50_imagenet_full.json | 6 ++++++ 3 files changed, 18 insertions(+) create mode 100644 brainscore_vision/models/scaling_models/region_layer_map/resnet18_imagenet_full.json create mode 100644 brainscore_vision/models/scaling_models/region_layer_map/resnet34_imagenet_full.json create mode 100644 brainscore_vision/models/scaling_models/region_layer_map/resnet50_imagenet_full.json 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 new file mode 100644 index 000000000..d06308cbb --- /dev/null +++ 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" +} \ No newline at end of file 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 new file mode 100644 index 000000000..a4e56916e --- /dev/null +++ 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" +} \ No newline at end of file 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 new file mode 100644 index 000000000..3e1342676 --- /dev/null +++ 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" +} \ No newline at end of file