diff --git a/brainscore_vision/models/resnext101_32x48d_wsl/__init__.py b/brainscore_vision/models/resnext101_32x48d_wsl/__init__.py index e566d9ea2..1b94e26b3 100644 --- a/brainscore_vision/models/resnext101_32x48d_wsl/__init__.py +++ b/brainscore_vision/models/resnext101_32x48d_wsl/__init__.py @@ -2,6 +2,6 @@ from brainscore_vision.model_helpers.brain_transformation import ModelCommitment from .model import get_model, get_layers -model_registry['resnext101_32x48d_wsl'] = lambda: ModelCommitment(identifier='resnext101_32x48d_wsl', - activations_model=get_model('resnext101_32x48d_wsl'), - layers=get_layers('resnext101_32x48d_wsl')) \ No newline at end of file +model_registry['resnext101_32x48d_wsl_TEST'] = lambda: ModelCommitment(identifier='resnext101_32x48d_wsl_TEST', + activations_model=get_model('resnext101_32x48d_wsl_TEST'), + layers=get_layers('resnext101_32x48d_wsl_TEST')) \ No newline at end of file diff --git a/brainscore_vision/models/resnext101_32x48d_wsl/model.py b/brainscore_vision/models/resnext101_32x48d_wsl/model.py index b3e08272b..2868433da 100644 --- a/brainscore_vision/models/resnext101_32x48d_wsl/model.py +++ b/brainscore_vision/models/resnext101_32x48d_wsl/model.py @@ -10,8 +10,8 @@ def get_model(name): - assert name == 'resnext101_32x48d_wsl' - model_identifier = "resnext101_32x48d_wsl" + assert name == 'resnext101_32x48d_wsl_TEST' + model_identifier = "resnext101_32x48d_wsl_TEST" model = torch.hub.load('facebookresearch/WSL-Images', model_identifier) preprocessing = functools.partial(load_preprocess_images, image_size=224) batch_size = {8: 32, 16: 16, 32: 8, 48: 4} @@ -22,7 +22,7 @@ def get_model(name): def get_layers(name): - assert name == 'resnext101_32x48d_wsl' + assert name == 'resnext101_32x48d_wsl_TEST' return (['conv1'] + # note that while relu is used multiple times, by default the last one will overwrite all previous ones [f"layer{block + 1}.{unit}.relu" diff --git a/brainscore_vision/models/resnext101_32x48d_wsl/requirements.txt b/brainscore_vision/models/resnext101_32x48d_wsl/requirements.txt index 5cad17cdd..a56666d38 100644 --- a/brainscore_vision/models/resnext101_32x48d_wsl/requirements.txt +++ b/brainscore_vision/models/resnext101_32x48d_wsl/requirements.txt @@ -1,3 +1,2 @@ torchvision -torch torch \ No newline at end of file diff --git a/brainscore_vision/models/resnext101_32x48d_wsl/test.py b/brainscore_vision/models/resnext101_32x48d_wsl/test.py index 8e223f69f..457f33a63 100644 --- a/brainscore_vision/models/resnext101_32x48d_wsl/test.py +++ b/brainscore_vision/models/resnext101_32x48d_wsl/test.py @@ -4,5 +4,5 @@ @pytest.mark.travis_slow def test_has_identifier(): - model = brainscore_vision.load_model('resnext101_32x48d_wsl') - assert model.identifier == 'resnext101_32x48d_wsl' \ No newline at end of file + model = brainscore_vision.load_model('resnext101_32x48d_wsl_TEST') + assert model.identifier == 'resnext101_32x48d_wsl_TEST' \ No newline at end of file