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

apex amp state dict #2555

Closed
ruotianluo opened this issue Jul 8, 2020 · 2 comments · Fixed by #2828
Closed

apex amp state dict #2555

ruotianluo opened this issue Jul 8, 2020 · 2 comments · Fixed by #2828
Labels
bug Something isn't working good first issue Good for newcomers help wanted Open to be worked on

Comments

@ruotianluo
Copy link
Contributor

🐛 Bug

https://github.com/PyTorchLightning/pytorch-lightning/blob/25ee51bc570503f331dceecc610d0eb355e22327/pytorch_lightning/trainer/training_io.py#L310

It seems for native amp support, the scalar state dict is saved. But for non native amp, the amp state dict is not saved?
@ruotianluo ruotianluo added bug Something isn't working help wanted Open to be worked on labels Jul 8, 2020
@ruotianluo ruotianluo changed the title amp state dict apex amp state dict Jul 8, 2020
@ruotianluo
Copy link
Contributor Author

anyone think it should be fixed?

@Borda
Copy link
Member

Borda commented Aug 4, 2020

@ruotianluo mind send a PR? 🐰

@Borda Borda added the good first issue Good for newcomers label Aug 4, 2020
@ruotianluo ruotianluo mentioned this issue Aug 4, 2020
Merged
7 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working good first issue Good for newcomers help wanted Open to be worked on
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

save apex scaler states ruotianluo/pytorch-lightning
2 participants
@Borda @ruotianluo