From 5adee377958a1ee9dcfd13876aa57994c731ebd3 Mon Sep 17 00:00:00 2001 From: tangwei Date: Thu, 14 May 2020 13:39:33 +0800 Subject: [PATCH] add paddle cloud run --- run.py | 1 + 1 file changed, 1 insertion(+) diff --git a/run.py b/run.py index ca89e88b..0a755cc2 100755 --- a/run.py +++ b/run.py @@ -151,6 +151,7 @@ def cluster_engine(args): cluster_envs = {} cluster_envs["train.trainer.trainer"] = trainer cluster_envs["train.trainer.engine"] = "cluster" + cluster_envs["train.trainer.threads"] = envs.get_runtime_environ("CPU_NUM") cluster_envs["train.trainer.device"] = args.device cluster_envs["train.trainer.platform"] = envs.get_platform() print("launch {} engine with cluster to with model: {}".format( -- GitLab