for startup base-model

d2be4f7a · xiexionghang · 38175506 · d2be4f7a
隐藏空白更改
内联并排

Showing with 5 addition and 3 deletion

paddle/fluid/train/custom_trainer/feed/process/learner_process.cc ...luid/train/custom_trainer/feed/process/learner_process.cc +5 -3

未找到文件。
--- 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);
+    if (_startup_dump_inference_base) {
-    environment->barrier(EnvironmentRole::WORKER); 
+        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("");