often 5 月之前
父节点
当前提交
528b40526d
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer.py

+ 1 - 1
recommend-model-produce/src/main/python/tools/static_ps_trainer.py

@@ -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(