diff --git a/brainscore_vision/models/inception_v4_pytorch/model.py b/brainscore_vision/models/inception_v4_pytorch/model.py index 0f4309b4a..f9d75221b 100644 --- a/brainscore_vision/models/inception_v4_pytorch/model.py +++ b/brainscore_vision/models/inception_v4_pytorch/model.py @@ -33,12 +33,16 @@ def get_model(name): def get_layers(name): assert name == 'inception_v4_pytorch' - layer_names = [] - - for name, module in MODEL.named_modules(): - layer_names.append(name) - - return layer_names[2:] + + 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'] + + return layers def get_bibtex(model_identifier):