From c24ce4a442222fc7819c2ac134d608b0f0407031 Mon Sep 17 00:00:00 2001
From: lilong12 <lilong12@baidu.com>
Date: Fri, 14 Feb 2020 10:56:31 +0800
Subject: [PATCH] fix the bug on the ouput of acc (#35)

---
 plsc/entry.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/plsc/entry.py b/plsc/entry.py
index 2667b0b..bdc8261 100644
--- a/plsc/entry.py
+++ b/plsc/entry.py
@@ -971,15 +971,15 @@ class Entry(object):
                     avg_lr = np.mean(local_train_info[1])
                     speed = nsamples / local_time
                     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(
                             pass_id,
                             batch_id,
                             avg_lr,
                             avg_loss,
                             speed,
-                            acc1,
-                            acc5))
+                            acc1[0],
+                            acc5[0]))
                     else:
                         logger.info("Pass:{} batch:{} lr:{:.8f} loss:{:.6f} "
                                     "qps:{:.2f}".format(pass_id,
-- 
GitLab