Skip to content

Commit

Permalink
Fix layer mapping
Browse files Browse the repository at this point in the history
  • Loading branch information
Ethan Pellegrini committed Jul 30, 2024
1 parent 48d8aee commit 72f989c
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions brainscore_vision/models/inception_v4_pytorch/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 72f989c

Please sign in to comment.