fix reset of Plot

carloslema · Mar 31, 2017 · 75cbdf9 · 75cbdf9
1 parent 55bf200
commit 75cbdf9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/python/paddle/v2/plot/plot.py b/python/paddle/v2/plot/plot.py
@@ -54,7 +54,10 @@ def plot(self):
         self.plt.gcf().clear()
 
     def reset(self):
-        self.__plot_data__ = []
+        for key in self.__plot_data__:
+            data = self.__plot_data__[key]
+            assert isinstance(data, PlotData)
+            data.reset()
 
 if __name__ == '__main__':
     title = "cost"
@@ -63,3 +66,7 @@ def reset(self):
     plot_test.append(title, 2, 2)
     for k, v in plot_test.__plot_data__.iteritems():
         print k, v.step, v.value
+    plot_test.reset()
+    for k, v in plot_test.__plot_data__.iteritems():
+        print k, v.step, v.value
+gg