丁云鹏 6 月之前
父節點
當前提交
ad1fa15239
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

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

@@ -331,8 +331,9 @@ class Main(object):
 
 
 
 
 if __name__ == "__main__":
 if __name__ == "__main__":
+
     paddle.enable_static()
     paddle.enable_static()
- 
+    paddle.base.framework.set_flags({'FLAGS_max_body_size': 2147483647})
     config = parse_args()
     config = parse_args()
     os.environ["CPU_NUM"] = str(config.get("runner.thread_num"))
     os.environ["CPU_NUM"] = str(config.get("runner.thread_num"))
     benchmark_main = Main(config)
     benchmark_main = Main(config)