fix: dbg
parent
df3e75dde4
commit
ccefbd7864
|
@ -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:
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue