Add GradientMachine::start/finish to API

demo/mnist/api_train.py

@@ -30,7 +30,12 @@ def main():
     updater = api.ParameterUpdater.createLocalUpdater(opt_config)
     assert isinstance(updater, api.ParameterUpdater)
     updater.init(m)
-    updater.startPass()
+    m.start()
+
+    for _ in xrange(100):
+        updater.startPass()
+
+    m.finish()
 
 
 if __name__ == '__main__':

paddle/api/GradientMachine.cpp

@@ -64,6 +64,10 @@ GradientMachine* GradientMachine::createByModelConfig(
   return GradientMachine::createFromPaddleModelPtr(confPtr, mode, types);
 }
 
+void GradientMachine::start() { m->machine->start(); }
+
+void GradientMachine::finish() { m->machine->finish(); }
+
 void GradientMachine::forward(const Arguments& inArgs,
                               Arguments* outArgs,
                               PassType passType) {

paddle/api/PaddleAPI.h

@@ -716,6 +716,13 @@ public:
       GradientMatchineCreateMode mode = CREATE_MODE_NORMAL,
       const std::vector<int>& parameterTypes = defaultParamTypes);
 
+  /**
+   * @brief finish
+   */
+  void finish();
+
+  void start();
+
   /**
    * The forward stage of GradientMachine.
    *
@@ -790,6 +797,8 @@ public:
 
   void startPass();
 
+  void finishPass();
+
 private:
   ParameterUpdaterPrivate* m;
 };

paddle/api/ParameterUpdater.cpp

@@ -33,3 +33,5 @@ void ParameterUpdater::init(const GradientMachine &gm) {
 }
 
 void ParameterUpdater::startPass() { m->updater->startPass(); }
+
+void ParameterUpdater::finishPass() {}