Fix the mertirc_list when \'use_mix==True\' (#529)

aa8e3c11 · Tingquan Gao · GitHub · 8c82d894 · aa8e3c11
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

tools/static/program.py tools/static/program.py +4 -2

未找到文件。
--- a/tools/static/program.py
+++ b/tools/static/program.py
@@ -503,8 +503,10 @@ def run(dataloader,
    ]
    topk_name = 'top{}'.format(config.topk)
    metric_list.insert(0, ("loss", fetchs["loss"][1]))
-    metric_list.insert(0, (topk_name, fetchs[topk_name][1]))
+    use_mix = config.get("use_mix", False) and mode == "train"
-    metric_list.insert(0, ("top1", fetchs["top1"][1]))
+    if not use_mix:
+        metric_list.insert(0, (topk_name, fetchs[topk_name][1]))
+        metric_list.insert(0, ("top1", fetchs["top1"][1]))
    metric_list = OrderedDict(metric_list)