From 72f989cac3d518d392e3b52ced39ddb4e583912e Mon Sep 17 00:00:00 2001 From: Ethan Pellegrini Date: Mon, 29 Jul 2024 23:00:45 -0400 Subject: [PATCH] Fix layer mapping --- .../models/inception_v4_pytorch/model.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/brainscore_vision/models/inception_v4_pytorch/model.py b/brainscore_vision/models/inception_v4_pytorch/model.py index f9d75221b..b3f059a8a 100644 --- a/brainscore_vision/models/inception_v4_pytorch/model.py +++ b/brainscore_vision/models/inception_v4_pytorch/model.py @@ -33,14 +33,14 @@ def get_model(name): def get_layers(name): assert name == 'inception_v4_pytorch' - - layers = ['Conv2d_1a_3x3'] + - ['Mixed_3a'] + - ['Mixed_4a'] + - [f'Mixed_5{i}' for i in ['a', 'b', 'c', 'd', 'e']] + - [f'Mixed_6{i}' for i in ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']] + - [f'Mixed_7{i}' for i in ['a', 'b', 'c', 'd']] + - ['global_pool'] + layers = [] + layers += ['Conv2d_1a_3x3'] + layers += ['Mixed_3a'] + layers += ['Mixed_4a'] + layers += [f'Mixed_5{i}' for i in ['a', 'b', 'c', 'd', 'e']] + layers += [f'Mixed_6{i}' for i in ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']] + layers += [f'Mixed_7{i}' for i in ['a', 'b', 'c', 'd']] + layers += ['global_pool'] return layers