pull/1989/head
gaotingquan 2022-05-18 06:27:02 +00:00 committed by Tingquan Gao
parent df3e75dde4
commit ccefbd7864
2 changed files with 2 additions and 2 deletions

View File

@ -21,9 +21,9 @@ import paddle.nn.functional as F
class Topk(object):
def __init__(self, topk=1, class_id_map_file=None, delimiter=None):
assert isinstance(topk, (int, ))
self.class_id_map = self.parse_class_id_map(class_id_map_file)
self.topk = topk
self.delimiter = delimiter if delimiter is not None else " "
self.class_id_map = self.parse_class_id_map(class_id_map_file)
def parse_class_id_map(self, class_id_map_file):
if class_id_map_file is None:

View File

@ -177,7 +177,7 @@ class Engine(object):
self.eval_metric_func = None
elif self.eval_mode == "retrieval":
if "Metric" in self.config and "Eval" in self.config["Metric"]:
metric_config = metric_config["Metric"]["Eval"]
metric_config = self.config["Metric"]["Eval"]
else:
metric_config = [{"name": "Recallk", "topk": (1, 5)}]
self.eval_metric_func = build_metrics(metric_config)