Skip to content

Commit

Permalink
add alexnet_4_all to models
Browse files Browse the repository at this point in the history
  • Loading branch information
Jenkins committed Dec 2, 2024
1 parent 9da6818 commit 86d7a97
Show file tree
Hide file tree
Showing 6 changed files with 2,922 additions and 0 deletions.
10 changes: 10 additions & 0 deletions brainscore_vision/models/alexnet_4_all/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@

from brainscore_vision import model_registry
from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
from .model import get_model, MODEL_CONFIGS, get_layers
network = 'alexnet'
for keyword in ['imagenet_trained', 'less_variation', 'no_variation', 'z_axis', 'wo_shading','wo_shadows', 'ambient', 'textures', 'no_specular']:
for iteration in range(1,6):
model_registry[f'{network}_{keyword}_iteration={iteration}'] = lambda: ModelCommitment(identifier=f'{network}_{keyword}_iteration={iteration}', activations_model=get_model('{network}_{keyword}_iteration={iteration}'), layers=get_layers(f'{network}_{keyword}_iteration={iteration}'))


Loading

0 comments on commit 86d7a97

Please sign in to comment.