提交 3d7e5ac0 编写于 作者: G guru4elephant

refine quick start example

上级 1194d1a0
...@@ -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.
先完成此消息的编辑!
想要评论请 注册