From 93edc0cf530a18dee5a50e91fe61a2dd05770660 Mon Sep 17 00:00:00 2001
From: "zuchen.wang" <zuchen.wang@vipshop.com>
Date: Wed, 10 Nov 2021 17:01:27 +0800
Subject: [PATCH] change evaluator
---
projects/FastShoe/fastshoe/trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/projects/FastShoe/fastshoe/trainer.py b/projects/FastShoe/fastshoe/trainer.py
index 5660a7f..16e86fd 100644
--- a/projects/FastShoe/fastshoe/trainer.py
+++ b/projects/FastShoe/fastshoe/trainer.py
@@ -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)