未验证 提交 aa8e3c11 编写于 作者: T Tingquan Gao 提交者: GitHub

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

上级 8c82d894
......@@ -503,6 +503,8 @@ def run(dataloader,
]
topk_name = 'top{}'.format(config.topk)
metric_list.insert(0, ("loss", fetchs["loss"][1]))
use_mix = config.get("use_mix", False) and mode == "train"
if not use_mix:
metric_list.insert(0, (topk_name, fetchs[topk_name][1]))
metric_list.insert(0, ("top1", fetchs["top1"][1]))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册