丁云鹏 5 місяців тому
батько
коміт
912e2b8b67

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

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