change relative module

3 years ago · 6e5d97e9ac
1 changed files with 3 additions and 3 deletions
--- a/distill_and_select.py
+++ b/distill_and_select.py
@ -18,9 +18,9 @@ from torchvision import transforms
 from towhee.operator.base import NNOperator
 from towhee import register
 from PIL import Image as PILImage
-from model.feature_extractor import FeatureExtractor
-from model.students import FineGrainedStudent, CoarseGrainedStudent
-from model.selector import SelectorNetwork
+from .model.feature_extractor import FeatureExtractor
+from .model.students import FineGrainedStudent, CoarseGrainedStudent
+from .model.selector import SelectorNetwork
 from torch import nn