未验证 提交 d49e1296 编写于 作者: L littletomatodonkey 提交者: GitHub

Update trainer.py

上级 eef713f8
......@@ -236,7 +236,7 @@ class Trainer(object):
model_name=self.config["Arch"]["name"],
prefix="best_model")
logger.info("[Eval][Epoch {}][best metric: {}]".format(
epoch_id, acc))
epoch_id, best_metric["metric"]))
self.model.train()
# save model
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册