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

change evaluator

pull/608/head
zuchen.wang 2021-11-10 17:01:27 +08:00
parent f8ae876c8c
commit 93edc0cf53
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,4 +86,4 @@ class PairTrainer(DefaultTrainer):
@classmethod @classmethod
def build_evaluator(cls, cfg, dataset_name, output_dir=None): def build_evaluator(cls, cfg, dataset_name, output_dir=None):
data_loader = cls.build_test_loader(cfg, dataset_name) data_loader = cls.build_test_loader(cfg, dataset_name)
return data_loader, PairDistanceEvaluator(cfg, output_dir) return data_loader, PairScoreEvaluator(cfg, output_dir)