From 59f4df30bce38bc4bba27b6ed2e84bd420d86275 Mon Sep 17 00:00:00 2001 From: tangwei Date: Mon, 20 Apr 2020 17:16:48 +0800 Subject: [PATCH] fix bug --- fleetrec/core/trainers/cluster_trainer.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/fleetrec/core/trainers/cluster_trainer.py b/fleetrec/core/trainers/cluster_trainer.py index 67dab613..1682a3da 100644 --- a/fleetrec/core/trainers/cluster_trainer.py +++ b/fleetrec/core/trainers/cluster_trainer.py @@ -43,12 +43,8 @@ class ClusterTrainer(TranspileTrainer): self.regist_context_processor('terminal_pass', self.terminal) def build_strategy(self): - mode = envs.get_global_env("train.strategy.mode") - - if mode is None: - mode = envs.get_runtime_envion("train.strategy.mode") - - assert mode is not None + mode = envs.get_runtime_envion("trainer.strategy") + assert mode in ["async", "geo", "sync", "half_async"] strategy = None -- GitLab