fix wrong import

trane293 · Sep 11, 2020 · 7190256 · 7190256
1 parent 2cf1fdc
commit 7190256
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/dataloader.py b/modules/dataloader.py
@@ -215,7 +215,8 @@ def standardize(images, findMeanVarOnly=True, saveDump=None, applyToTest=None):
     # takes a dictionary
     if applyToTest != None:
         logger.info('Applying to test data using provided values')
-        from training_helpers import apply_mean_std
+        # from training_helpers import apply_mean_std
+        from .helpers import apply_mean_std
         images = apply_mean_std(images, applyToTest)
         return images