@@ -225,7 +225,7 @@ class Main(object):
fleet.barrier_worker()
# 2. 获取主程序
- main_program = fleet.main_program
+ main_program = paddle.static.default_main_program()
# 3. 使用 paddle.static.save_inference_model 替代 fleet.save_inference_model
paddle.static.save_inference_model(