提交 6785af02 编写于 作者: C chengmo

fix

上级 2f1f76f3
...@@ -62,9 +62,6 @@ class SingleTrainer(TranspileTrainer): ...@@ -62,9 +62,6 @@ class SingleTrainer(TranspileTrainer):
context['status'] = 'train_pass' context['status'] = 'train_pass'
def dataloader_train(self, context): def dataloader_train(self, context):
self.model.custom_preprocess()
reader = self._get_dataloader() reader = self._get_dataloader()
epochs = envs.get_global_env("train.epochs") epochs = envs.get_global_env("train.epochs")
...@@ -105,10 +102,6 @@ class SingleTrainer(TranspileTrainer): ...@@ -105,10 +102,6 @@ class SingleTrainer(TranspileTrainer):
context['status'] = 'infer_pass' context['status'] = 'infer_pass'
def dataset_train(self, context): def dataset_train(self, context):
# run startup program at once
self._exe.run(fluid.default_startup_program())
self.model.custom_preprocess()
dataset = self._get_dataset() dataset = self._get_dataset()
epochs = envs.get_global_env("train.epochs") epochs = envs.get_global_env("train.epochs")
......
...@@ -18,9 +18,6 @@ def engine_registry(): ...@@ -18,9 +18,6 @@ def engine_registry():
cpu["TRANSPILER"]["SINGLE"] = single_engine cpu["TRANSPILER"]["SINGLE"] = single_engine
cpu["TRANSPILER"]["LOCAL_CLUSTER"] = local_cluster_engine cpu["TRANSPILER"]["LOCAL_CLUSTER"] = local_cluster_engine
cpu["TRANSPILER"]["CLUSTER"] = cluster_engine cpu["TRANSPILER"]["CLUSTER"] = cluster_engine
cpu["TRANSPILER"]["TDM_SINGLE"] = tdm_single_engine
cpu["TRANSPILER"]["TDM_LOCAL_CLUSTER"] = tdm_local_cluster_engine
cpu["TRANSPILER"]["TDM_CLUSTER"] = tdm_cluster_engine
cpu["PSLIB"]["SINGLE"] = local_mpi_engine cpu["PSLIB"]["SINGLE"] = local_mpi_engine
cpu["PSLIB"]["LOCAL_CLUSTER"] = local_mpi_engine cpu["PSLIB"]["LOCAL_CLUSTER"] = local_mpi_engine
cpu["PSLIB"]["CLUSTER"] = cluster_mpi_engine cpu["PSLIB"]["CLUSTER"] = cluster_mpi_engine
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册