From 6785af02f398f074c0358dfcd6855b9bbfbd51bb Mon Sep 17 00:00:00 2001 From: chengmo Date: Wed, 6 May 2020 23:11:52 +0800 Subject: [PATCH] fix --- fleet_rec/core/trainers/single_trainer.py | 7 ------- fleet_rec/run.py | 3 --- 2 files changed, 10 deletions(-) diff --git a/fleet_rec/core/trainers/single_trainer.py b/fleet_rec/core/trainers/single_trainer.py index 74cb7620..2ad99f95 100644 --- a/fleet_rec/core/trainers/single_trainer.py +++ b/fleet_rec/core/trainers/single_trainer.py @@ -62,9 +62,6 @@ class SingleTrainer(TranspileTrainer): context['status'] = 'train_pass' def dataloader_train(self, context): - - self.model.custom_preprocess() - reader = self._get_dataloader() epochs = envs.get_global_env("train.epochs") @@ -105,10 +102,6 @@ class SingleTrainer(TranspileTrainer): context['status'] = 'infer_pass' 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() epochs = envs.get_global_env("train.epochs") diff --git a/fleet_rec/run.py b/fleet_rec/run.py index 51ba0499..b10428c8 100644 --- a/fleet_rec/run.py +++ b/fleet_rec/run.py @@ -18,9 +18,6 @@ def engine_registry(): cpu["TRANSPILER"]["SINGLE"] = single_engine cpu["TRANSPILER"]["LOCAL_CLUSTER"] = local_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"]["LOCAL_CLUSTER"] = local_mpi_engine cpu["PSLIB"]["CLUSTER"] = cluster_mpi_engine -- GitLab