Skip to content

[Optimizer] polish the init method of ColoOptimizer#1310

Merged
feifeibear merged 14 commits intohpcaitech:mainfrom
feifeibear:checkpoint/optim
Jul 14, 2022
Merged

[Optimizer] polish the init method of ColoOptimizer#1310
feifeibear merged 14 commits intohpcaitech:mainfrom
feifeibear:checkpoint/optim

Conversation

@feifeibear
Copy link
Copy Markdown
Contributor

No description provided.

@feifeibear feifeibear requested a review from 1SAA July 14, 2022 07:40
@@ -70,6 +71,6 @@ def load_checkpoint(dire,

# reset tensors to original dist spec.
with DistSpecManager.no_grad():
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

This sentence can be polished out.

@feifeibear feifeibear merged commit 9f10524 into hpcaitech:main Jul 14, 2022
@feifeibear feifeibear deleted the checkpoint/optim branch July 14, 2022 08:37
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