From 7c0703f21047ba413662e6f08db84228e7567aa4 Mon Sep 17 00:00:00 2001 From: chengmo Date: Wed, 6 May 2020 23:56:09 +0800 Subject: [PATCH] fix --- fleet_rec/core/trainers/tdm_cluster_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fleet_rec/core/trainers/tdm_cluster_trainer.py b/fleet_rec/core/trainers/tdm_cluster_trainer.py index ba7bd72b..2b87b3c8 100644 --- a/fleet_rec/core/trainers/tdm_cluster_trainer.py +++ b/fleet_rec/core/trainers/tdm_cluster_trainer.py @@ -25,7 +25,7 @@ from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distribu from paddle.fluid.incubate.fleet.base.role_maker import PaddleCloudRoleMaker from fleetrec.core.utils import envs -from fleetrec.core.trainers.transpiler_trainer import TranspileTrainer +from fleetrec.core.trainers.cluster_trainer import ClusterTrainer logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s") @@ -34,7 +34,7 @@ logger.setLevel(logging.INFO) special_param = ["TDM_Tree_Travel", "TDM_Tree_Layer", "TDM_Tree_Info"] -class TDMClusterTrainer(TranspileTrainer): +class TDMClusterTrainer(ClusterTrainer): def server(self, context): namespace = "train.startup" init_model_path = envs.get_global_env( -- GitLab