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(