update import

mit-han-lab · Jul 19, 2023 · 101fa22 · 101fa22
1 parent a5381c1
commit 101fa22
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/ofa/imagenet_classification/data_providers/imagenet.py b/ofa/imagenet_classification/data_providers/imagenet.py
@@ -42,7 +42,7 @@ def __init__(
 
         self._valid_transform_dict = {}
         if not isinstance(self.image_size, int):
-            from ofa.utils.my_dataloader import MyDataLoader
+            from ofa.utils.my_dataloader.my_data_loader import MyDataLoader
 
             assert isinstance(self.image_size, list)
             self.image_size.sort()  # e.g., 160 -> 224

diff --git a/ofa/utils/my_dataloader/__init__.py b/ofa/utils/my_dataloader/__init__.py
@@ -1,4 +1,2 @@
-from .my_data_loader import *
-from .my_data_worker import *
 from .my_distributed_sampler import *
 from .my_random_resize_crop import *