From 452caa0954a855df13ef72bf65f5a3872eaf5ee7 Mon Sep 17 00:00:00 2001
From: zhangchao <zhangchao41@baidu.com>
Date: Tue, 9 Jan 2018 19:08:56 +0800
Subject: [PATCH] Update plotlog.py

---
 benchmark/paddle/image/plotlog.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/benchmark/paddle/image/plotlog.py b/benchmark/paddle/image/plotlog.py
index 4ac8edfea..41f535d98 100644
--- a/benchmark/paddle/image/plotlog.py
+++ b/benchmark/paddle/image/plotlog.py
@@ -79,10 +79,10 @@ def plot_metric(metric,
     plt.figure()
     plt.title(graph_title)
     if line_num == 1:
-        plt.plot(batch_id, metric, line_style, line_label)
+        plt.plot(batch_id, metric, line_style, label=line_label)
     else:
         for i in range(line_num):
-            plt.plot(batch_id, metric[i], line_style[i], line_label[i])
+            plt.plot(batch_id, metric[i], line_style[i], label=line_label[i])
     plt.xlabel('batch')
     plt.ylabel(graph_title)
     plt.legend()
@@ -102,12 +102,12 @@ def main():
     accuracy_sample = sample(accuracy, args.sample_rate)
 
     plot_metric(loss_sample, batch_sample, 'loss', line_label='loss')
-    plot_metric(accuracy_sample,
-                batch_sample,
-                'accuracy',
-                line_style='g-',
-                line_label='accuracy')
-
+    plot_metric(
+        accuracy_sample,
+        batch_sample,
+        'accuracy',
+        line_style='g-',
+        line_label='accuracy')
 
 if __name__ == '__main__':
     main()
-- 
GitLab