fix ordered_save func (#10896)

RevL147 · luyuxiang · web-flow · commit 2ae914eae9cc · 2025-08-05T12:08:30.000+08:00
Co-authored-by: luyuxiang &lt;luyuxiang@YuxiangdeMacBook-Pro.local&gt;
diff --git a/paddlenlp/trainer/trainer.py b/paddlenlp/trainer/trainer.py
@@ -2657,7 +2657,7 @@ def _filter_moe_no_sync_optimizer_params(self):
                         filter_optimzier_state_dict[op_k] = op_v
         return filter_optimzier_state_dict
 
-    def _ordered_save(self, state_dict, save_path):
+    def _ordered_save(self, state_dict, save_path, signal_path=None):
         group_size = self.args.ordered_save_group_size
         hcg = fleet.get_hybrid_communicate_group()
         if hcg.get_sharding_parallel_world_size() > 1 or hcg.get_model_parallel_world_size() <= 1:
@@ -2677,6 +2677,10 @@ def _ordered_save(self, state_dict, save_path):
                 paddle.save(state_dict, save_path)
             dist.barrier(mp_group)
 
+        if signal_path is not None:
+            with open(signal_path, mode="w+") as f:
+                f.write("1")
+
     def _save_checkpoint(self, model, metrics=None):
         # assert unwrap_model(model) is self.model, "internal model should be a reference to self.model"
         if self.args.enable_zero_cost_checkpoint:
