logo

towhee
/
distill-and-select
repo-copy-icon

copied

Browse Source

change relative module

main
ChengZi 3 years ago
parent
9401a60fb6
commit
6e5d97e9ac
1 changed files with 3 additions and 3 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 6
      distill_and_select.py

6
distill_and_select.py
View File

@ -18,9 +18,9 @@ from torchvision import transforms
from towhee.operator.base import NNOperator from towhee.operator.base import NNOperator
from towhee import register from towhee import register
from PIL import Image as PILImage 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 from torch import nn

Write Preview
Loading…
Cancel
Save