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

View File

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