diff --git a/brainscore_vision/models/fixres_resnext101_32x48d_wsl/model.py b/brainscore_vision/models/fixres_resnext101_32x48d_wsl/model.py index 594a627a8..b9d82f402 100644 --- a/brainscore_vision/models/fixres_resnext101_32x48d_wsl/model.py +++ b/brainscore_vision/models/fixres_resnext101_32x48d_wsl/model.py @@ -1,7 +1,7 @@ from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper from fixres.hubconf import load_state_dict_from_url from fixres.transforms_v2 import get_transforms -from model_helpers.activations.pytorch import load_images +from brainscore_vision.model_helpers.activations.pytorch import load_images import numpy as np from importlib import import_module import ssl @@ -54,4 +54,4 @@ def get_layers(name): # note that while relu is used multiple times, by default the last one will overwrite all previous ones [f"layer{block + 1}.{unit}.relu" for block, block_units in enumerate([3, 4, 23, 3]) for unit in range(block_units)] + - ['avgpool']) \ No newline at end of file + ['avgpool'])