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

qizhaoaoe · Feb 14, 2020 · c24ce4a · c24ce4a
1 parent 92608de
commit c24ce4a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plsc/entry.py b/plsc/entry.py
@@ -971,15 +971,15 @@ def train(self):
                     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,