mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

optimize import

pull/608/head
zuchen.wang 2021-11-18 17:24:30 +08:00
parent 270520c864
commit 2cdc57f5d2
1 changed files with 2 additions and 2 deletions
projects/FastShoe/fastshoe

4
projects/FastShoe/fastshoe/trainer.py
View File

@ -14,8 +14,8 @@ from fastreid.utils import comm
from fastreid.data.transforms import build_transforms
from fastreid.data.build import build_reid_train_loader, build_reid_test_loader
from fastreid.evaluation import PairScoreEvaluator, PairDistanceEvaluator
from projects.FastShoe.fastshoe.data import PairDataset
# ensure custom datasets are registered
import projects.FastShoe.fastshoe.data
class PairTrainer(DefaultTrainer):