diff --git a/paddle/fluid/train/custom_trainer/feed/process/learner_process.cc b/paddle/fluid/train/custom_trainer/feed/process/learner_process.cc index a2932d8a2309bd344ee90215c9d562768b58670d..c42c1c5977916958c843c66b629fb5ff1b796397 100755 --- a/paddle/fluid/train/custom_trainer/feed/process/learner_process.cc +++ b/paddle/fluid/train/custom_trainer/feed/process/learner_process.cc @@ -198,9 +198,11 @@ int LearnerProcess::run() { environment->barrier(EnvironmentRole::WORKER); //判断是否先dump出base TODO - wait_save_model(epoch_id, ModelSaveWay::ModelSaveInferenceBase, _startup_dump_inference_base); - environment->barrier(EnvironmentRole::WORKER); - + if (_startup_dump_inference_base) { + wait_save_model(epoch_id, ModelSaveWay::ModelSaveInferenceBase, _startup_dump_inference_base); + environment->barrier(EnvironmentRole::WORKER); + } + while (true) { epoch_accessor->next_epoch(); _context_ptr->monitor_ssm.str("");