Skip to content

document about engine.load_checkpoint() - #1457

Merged
tjruwase merged 1 commit into
deepspeedai:masterfrom
skpig:document_load_checkpoint
Oct 18, 2021
Merged

tjruwase merged 1 commit into
deepspeedai:masterfrom
skpig:document_load_checkpoint

Conversation

@skpig

@skpig skpig commented Oct 17, 2021

Copy link
Copy Markdown
Contributor

About issue #1394. Add document about engine.load_checkpoint() explaining the limitation under ZeRO3. @stas00

@skpig

skpig commented Oct 17, 2021

Copy link
Copy Markdown
Contributor Author

It is my first time to make a PR. If I did anything wrong, please forgive me. It will be appreciated if anyone could give me some advice.

@stas00

stas00 commented Oct 17, 2021

Copy link
Copy Markdown
Collaborator

Looks great to me, @skpig!

Could you please test if the problem occurs only with ZeRO3? It's most likely the case, but it'd be good to validate that ZeRO2 doesn't have this limitation.

@skpig

skpig commented Oct 18, 2021

Copy link
Copy Markdown
Contributor Author

I just checked. The issue doesn't occur under ZeRO2.

@stas00

stas00 commented Oct 18, 2021

Copy link
Copy Markdown
Collaborator

Thank you for doing the work, @skpig!

@tjruwase, this is good to be merged then, IMHO.

@tjruwase

Copy link
Copy Markdown
Contributor

Thanks @skpig and @stas00.

@tjruwase
tjruwase merged commit b4e5826 into deepspeedai:master Oct 18, 2021
Dipet pushed a commit to Dipet/DeepSpeed that referenced this pull request Oct 26, 2021
Co-authored-by: skpig <1900012999@pku.edu.cn>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants