Fix imports

brain-score · Jul 30, 2024 · 961a559 · 961a559
1 parent bca335f
commit 961a559
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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'])
+            ['avgpool'])