Wait for reading data.

8 years ago · 27d87db6a0
parent 677c79b6a1
commit 27d87db6a0
2 changed files with 3 additions and 1 deletions
--- 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()


--- 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(); }