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

support amp with fused optimizer #379

Open

wangraying opened this issue Nov 16, 2021 · 2 comments

Member

wangraying commented Nov 16, 2021 •

edited

Loading

torch native amp + apex amp

windj007 commented Feb 26, 2022

Hi! Any updates on this issue? Is it possible to use fuse_optimizer with torch.cuda.amp.GradScaler?

Member Author

wangraying commented Feb 27, 2022 •

edited

Loading

Not yet. We may solve this issue in Q2 this year, hopefully. There is a previous draft related to this issue master...fuse-opt-fp16.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment