diff --git a/paddle/fluid/distributed/fleet_executor/dist_model.cc b/paddle/fluid/distributed/fleet_executor/dist_model.cc index 27a4b0792c2b158a5f583dcdb322db6c6a7fa9ca..89edc3a60c882ac7cab49f5cd6f0216ac9d33726 100644 --- a/paddle/fluid/distributed/fleet_executor/dist_model.cc +++ b/paddle/fluid/distributed/fleet_executor/dist_model.cc @@ -436,7 +436,11 @@ bool DistModel::PrepareFleetExe() { for (int i = 0; i < config_.nranks; ++i) { RankInfo *rank_info = executor_desc_.add_cluster_info(); rank_info->set_rank(i); - rank_info->set_ip_port(config_.trainer_endpoints[i]); + if (config_.nranks == 1) { + rank_info->set_ip_port(""); + } else { + rank_info->set_ip_port(config_.trainer_endpoints[i]); + } id_to_rank.insert({i, i}); } fleet_exe.reset(new FleetExecutor(executor_desc_));