未验证 提交 9f0d73a3 编写于 作者: D Dong Daxiang 提交者: GitHub

Merge pull request #13 from guru4elephant/master

refine quick start example
...@@ -88,8 +88,12 @@ step_i = 0 ...@@ -88,8 +88,12 @@ step_i = 0
while not trainer.stop(): while not trainer.stop():
step_i += 1 step_i += 1
print("batch %d start train" % (step_i)) print("batch %d start train" % (step_i))
trainer.train_inner_loop(reader) trainer.run(feed=data, fetch=[])
trainer.save_inference_program(output_folder) 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: On FL Server Node, a training script is defined as follows:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册