From 27d87db6a0f937a7fa22b03e3d18844f894698e1 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Wed, 21 Dec 2016 12:54:11 +0800
Subject: [PATCH] Wait for reading data.

---
 demo/mnist/api_train.py         | 2 ++
 paddle/api/ParameterUpdater.cpp | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/demo/mnist/api_train.py b/demo/mnist/api_train.py
index b061cfb2b8f..59043ce6c42 100644
--- a/demo/mnist/api_train.py
+++ b/demo/mnist/api_train.py
@@ -35,6 +35,8 @@ def main():
     for _ in xrange(100):
         updater.startPass()
 
+        updater.finishPass()
+
     m.finish()
 
 
diff --git a/paddle/api/ParameterUpdater.cpp b/paddle/api/ParameterUpdater.cpp
index 3b626c05071..4edec78b4a3 100644
--- a/paddle/api/ParameterUpdater.cpp
+++ b/paddle/api/ParameterUpdater.cpp
@@ -34,4 +34,4 @@ void ParameterUpdater::init(const GradientMachine &gm) {
 
 void ParameterUpdater::startPass() { m->updater->startPass(); }
 
-void ParameterUpdater::finishPass() {}
+void ParameterUpdater::finishPass() { m->updater->finishPass(); }
-- 
GitLab