Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Merging optimizer states from different pipeline parallel size to resume training #38

Closed
xrsrke opened this issue Jan 23, 2024 · 0 comments
Closed

Merging optimizer states from different pipeline parallel size to resume training #38

xrsrke opened this issue Jan 23, 2024 · 0 comments
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed

Comments

@xrsrke
Copy link
Member

xrsrke commented Jan 23, 2024 •
edited
Loading

Suppose you start training with a pipeline parallel size of 4. We need to make it supports resuming training with a different pipeline parallel size, like 2, by merging optimizer states.
@xrsrke xrsrke added enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed labels Jan 23, 2024
@xrsrke xrsrke changed the title Merging optimizer states from different topologies to resume training Merging optimizer states from different pipeline parallel size to resume training Jan 24, 2024
@nopperl nopperl mentioned this issue Feb 15, 2024
Merged
@xrsrke xrsrke closed this as completed Feb 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@xrsrke