From ccefbd78649065f4f7fc865a1db159d4da2c50b8 Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Wed, 18 May 2022 06:27:02 +0000 Subject: [PATCH] fix: dbg --- ppcls/data/postprocess/topk.py | 2 +- ppcls/engine/engine.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/data/postprocess/topk.py b/ppcls/data/postprocess/topk.py index df027194..76772f56 100644 --- a/ppcls/data/postprocess/topk.py +++ b/ppcls/data/postprocess/topk.py @@ -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: diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 3e39daef..34cb4c4d 100644 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -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) -- GitLab