From 23e79ec597f69f74101cd09973c6f774bb7e239f Mon Sep 17 00:00:00 2001
From: dzhwinter <dzhwinter@gmail.com>
Date: Mon, 22 Jan 2018 23:42:05 -0800
Subject: [PATCH 1/2] mnist

---
 fluid/mnist.py         | 13 +++++++++++++
 fluid/requirements.txt |  3 +++
 2 files changed, 16 insertions(+)
 create mode 100644 fluid/requirements.txt

diff --git a/fluid/mnist.py b/fluid/mnist.py
index c7f5fa8..44caf76 100644
--- a/fluid/mnist.py
+++ b/fluid/mnist.py
@@ -10,6 +10,13 @@
 import paddle.v2.fluid as fluid
 import paddle.v2.fluid.profiler as profiler
 
+from visualdl import LogWriter
+
+logwriter = LogWriter("./tmp", sync_cycle=10)
+with logwriter.mode("train") as writer:
+    loss_scalar = writer.scalar("loss")
+    acc_scalar = writer.scalar("acc")
+
 SEED = 1
 DTYPE = "float32"
 
@@ -105,6 +112,7 @@ def run_benchmark(model, args):
         pr = cProfile.Profile()
         pr.enable()
     start_time = time.time()
+    step = 0
     # Input data
     images = fluid.layers.data(name='pixel', shape=[1, 28, 28], dtype=DTYPE)
     label = fluid.layers.data(name='label', shape=[1], dtype='int64')
@@ -161,6 +169,11 @@ def run_benchmark(model, args):
             end = time.time()
             loss = np.array(outs[0])
             acc = np.array(outs[1])
+
+            loss_scalar.add_record(step, loss)
+            acc_scalar.add_record(step, acc)
+            step += 1
+
             print("pass=%d, batch=%d, loss=%f, error=%f, elapse=%f" %
                   (pass_id, batch_id, loss, 1 - acc, (end - start) / 1000))
 
diff --git a/fluid/requirements.txt b/fluid/requirements.txt
new file mode 100644
index 0000000..837372f
--- /dev/null
+++ b/fluid/requirements.txt
@@ -0,0 +1,3 @@
+visualdl
+numpy
+paddlepaddle

From 65936df53d7d8e858d0b62378153657d9bad24c2 Mon Sep 17 00:00:00 2001
From: dzhwinter <dzhwinter@gmail.com>
Date: Mon, 22 Jan 2018 23:51:02 -0800
Subject: [PATCH 2/2] add

---
 fluid/mnist.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fluid/mnist.py b/fluid/mnist.py
index 44caf76..53fe435 100644
--- a/fluid/mnist.py
+++ b/fluid/mnist.py
@@ -12,7 +12,7 @@
 
 from visualdl import LogWriter
 
-logwriter = LogWriter("./tmp", sync_cycle=10)
+logwriter = LogWriter("./tmp", sync_cycle=100)
 with logwriter.mode("train") as writer:
     loss_scalar = writer.scalar("loss")
     acc_scalar = writer.scalar("acc")