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