From eeaf9942224254c6ab27fffe0cfd44d750f486ee Mon Sep 17 00:00:00 2001 From: tangwei Date: Mon, 20 Apr 2020 16:13:49 +0800 Subject: [PATCH] fix readme --- fleetrec/core/trainers/transpiler_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fleetrec/core/trainers/transpiler_trainer.py b/fleetrec/core/trainers/transpiler_trainer.py index ee8ea3f1..3f8e41e4 100644 --- a/fleetrec/core/trainers/transpiler_trainer.py +++ b/fleetrec/core/trainers/transpiler_trainer.py @@ -39,7 +39,7 @@ class TranspileTrainer(Trainer): namespace = "train.reader" inputs = self.model.get_inputs() - threads = envs.get_global_env("train.threads", None) + threads = int(envs.get_runtime_envion("trainer.threads")) batch_size = envs.get_global_env("batch_size", None, namespace) reader_class = envs.get_global_env("class", None, namespace) abs_dir = os.path.dirname(os.path.abspath(__file__)) -- GitLab