未验证 提交 c24ce4a4 编写于 作者: L lilong12 提交者: GitHub

fix the bug on the ouput of acc (#35)

上级 92608de7
...@@ -971,15 +971,15 @@ class Entry(object): ...@@ -971,15 +971,15 @@ class Entry(object):
avg_lr = np.mean(local_train_info[1]) avg_lr = np.mean(local_train_info[1])
speed = nsamples / local_time speed = nsamples / local_time
if self.calc_train_acc: if self.calc_train_acc:
logger.info("Pass:{} batch:%d lr:{:.8f} loss:{:.6f} " logger.info("Pass:{} batch:{} lr:{:.8f} loss:{:.6f} "
"qps:{:.2f} acc1:{:.6f} acc5:{:.6f}".format( "qps:{:.2f} acc1:{:.6f} acc5:{:.6f}".format(
pass_id, pass_id,
batch_id, batch_id,
avg_lr, avg_lr,
avg_loss, avg_loss,
speed, speed,
acc1, acc1[0],
acc5)) acc5[0]))
else: else:
logger.info("Pass:{} batch:{} lr:{:.8f} loss:{:.6f} " logger.info("Pass:{} batch:{} lr:{:.8f} loss:{:.6f} "
"qps:{:.2f}".format(pass_id, "qps:{:.2f}".format(pass_id,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册