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