Merge f6f56f0a7d into 4a68da72d5

trainer/train_agent.py

@@ -492,6 +492,4 @@ if __name__ == "__main__":
         else:
             rl_train_epoch(epoch, loader, len(loader), rollout_engine, ref_model, reward_model, 0, wandb, use_sglang = (args.rollout_engine == "sglang"))
 
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_distillation.py

@@ -248,6 +248,4 @@ if __name__ == "__main__":
             train_epoch(epoch, loader, len(loader), teacher_model, lm_config_student, 0, wandb, args.alpha, args.temperature)
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_dpo.py

@@ -228,6 +228,4 @@ if __name__ == "__main__":
             train_epoch(epoch, loader, len(loader), ref_model, lm_config, 0, wandb, args.beta)
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_full_sft.py

@@ -173,6 +173,4 @@ if __name__ == "__main__":
             train_epoch(epoch, loader, len(loader), 0, wandb)
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_grpo.py

@@ -334,6 +334,4 @@ if __name__ == "__main__":
             grpo_train_epoch(epoch, loader, len(loader), rollout_engine, ref_model, reward_model, 0, wandb, use_sglang = (args.rollout_engine == "sglang"))
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_lora.py

@@ -186,6 +186,4 @@ if __name__ == "__main__":
             train_epoch(epoch, loader, len(loader), lora_params, 0, wandb)
     
     # ========== 10. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_ppo.py

@@ -437,6 +437,4 @@ if __name__ == "__main__":
             ppo_train_epoch(epoch, loader, len(loader), rollout_engine, ref_model, actor_scheduler, critic_scheduler, reward_model, 0, wandb, use_sglang = (args.rollout_engine == "sglang"))
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()

trainer/train_pretrain.py

@@ -172,6 +172,4 @@ if __name__ == "__main__":
             train_epoch(epoch, loader, len(loader), 0, wandb)
     
     # ========== 9. 清理分布进程 ==========
-    if dist.is_initialized():
-        dist.barrier()
-        dist.destroy_process_group()
+    if dist.is_initialized(): dist.destroy_process_group()