fix the rdzv name to check node.

intelligent-machine-learning · May 31, 2024 · 368b2e2 · 368b2e2
1 parent 27c965f
commit 368b2e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlrover/python/elastic_agent/torch/training.py b/dlrover/python/elastic_agent/torch/training.py
@@ -1049,7 +1049,7 @@ def _create_check_agent(
 
  spec = _create_worker_spec(
  node_rank=node_rank,
- rdzv_name=RendezvousName.ELASTIC_TRAINING,
+ rdzv_name=rdzv_name,
  config=config,
  entrypoint=entrypoint,
  args=args,