Skip to content

Conversation

@d4l3k
Copy link
Member

@d4l3k d4l3k commented Jan 31, 2025

This updates timeouts for train_ddp.py so it recovers fast and some documentation to make it clearer what each option does.

Test plan:

torchft_lighthouse --min_replicas 2 --join_timeout_ms 10000
torchx run -- --replicas 10
  • killing workers via dashboard

@d4l3k d4l3k requested review from H-Huang, c-p-i-o and fegin January 31, 2025 19:51
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Jan 31, 2025
Copy link
Contributor

@H-Huang H-Huang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@d4l3k d4l3k merged commit 6e5dcbd into main Jan 31, 2025
6 checks passed
@d4l3k d4l3k deleted the d4l3k/ddp_timeouts branch January 31, 2025 23:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants