Home Explore Help
Sign In
algorithm
/
recommend-model
15
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

test

often 5 months ago
parent
f8f64c318c
commit
528b40526d
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  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
View File 

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