From 6e5d97e9acb4a80df10d9fb90d2703701aaad594 Mon Sep 17 00:00:00 2001 From: ChengZi Date: Wed, 6 Jul 2022 15:32:39 +0800 Subject: [PATCH] change relative module --- distill_and_select.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/distill_and_select.py b/distill_and_select.py index 07cc379..52c3bea 100644 --- 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