提交 eeaf9942 编写于 作者: T tangwei

fix readme

上级 1096f044
...@@ -39,7 +39,7 @@ class TranspileTrainer(Trainer): ...@@ -39,7 +39,7 @@ class TranspileTrainer(Trainer):
namespace = "train.reader" namespace = "train.reader"
inputs = self.model.get_inputs() 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) batch_size = envs.get_global_env("batch_size", None, namespace)
reader_class = envs.get_global_env("class", None, namespace) reader_class = envs.get_global_env("class", None, namespace)
abs_dir = os.path.dirname(os.path.abspath(__file__)) abs_dir = os.path.dirname(os.path.abspath(__file__))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册