From b45368e803bc551aedd04b4d5425f9019159d369 Mon Sep 17 00:00:00 2001 From: tangwei Date: Wed, 15 Apr 2020 19:41:29 +0800 Subject: [PATCH] rename upper to lowe --- fleetrec/run.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fleetrec/run.py b/fleetrec/run.py index db1b56f2..0a052072 100644 --- a/fleetrec/run.py +++ b/fleetrec/run.py @@ -45,13 +45,13 @@ if __name__ == "__main__": if not os.path.exists(args.model) or not os.path.isfile(args.model): raise ValueError("argument model: {} error, must specify a existed yaml file".format(args.model)) - if args.engine == "Single": + if args.engine.upper == "SINGLE": print("use SingleTraining to run model: {}".format(args.model)) single_envs = {} single_envs["train.trainer"] = "SingleTraining" single_engine(single_envs, args.model) - elif args.engine == "LocalCluster": + elif args.engine.upper == "LOCAL_CLUSTER": print("use 1X1 ClusterTraining at localhost to run model: {}".format(args.model)) cluster_envs = {} @@ -63,7 +63,7 @@ if __name__ == "__main__": cluster_envs["train.strategy.mode"] = "async" local_cluster_engine(cluster_envs, args.model) - elif args.engine == "LocalMPI": + elif args.engine.upper == "LOCAL_MPI": print("use 1X1 MPI ClusterTraining at localhost to run model: {}".format(args.model)) cluster_envs = {} -- GitLab