[fix][RL] Remove the incorrect barrier in init_weights_update_group (#5914)

This commit is contained in:
Zilin Zhu
2025-05-15 10:15:21 +08:00
committed by GitHub
parent f3bf611054
commit 44a3783d13
2 changed files with 2 additions and 4 deletions

View File

@@ -631,7 +631,6 @@ class ModelRunner:
rank=rank,
group_name=group_name,
)
dist.barrier(group=self._model_update_group, device_ids=[rank])
return True, "Succeeded to initialize custom process group."
except Exception as e:
message = f"Failed to initialize custom process group: {e}."