Merge pull request #13 from guru4elephant/master

refine quick start example
PaddlePaddle · Nov 13, 2019 · 9f0d73a · 9f0d73a
2 parents 6970e81 + 3d7e5ac
commit 9f0d73a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/docs/source/md/quick_start.md b/docs/source/md/quick_start.md
@@ -88,8 +88,12 @@ step_i = 0
 while not trainer.stop():
     step_i += 1
     print("batch %d start train" % (step_i))
-    trainer.train_inner_loop(reader)
-    trainer.save_inference_program(output_folder)
+    trainer.run(feed=data, fetch=[])
+    if trainer_id == 0:
+        print("start saving model")
+        trainer.save_inference_program(output_folder)
+    if step_i >= 100:
+       break
 ```
 
 On FL Server Node, a training script is defined as follows: