Go
pytorch
2e761498
- Revert "`GradScaler` recomputes `optimizer_state["found_inf_per_device"]` before `optimizer.step` (#97415) (#97886)" (#98624)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
Revert "`GradScaler` recomputes `optimizer_state["found_inf_per_device"]` before `optimizer.step` (#97415) (#97886)" (#98624) This reverts commit 57cd423a4fd119a7ecf944cd7aa43e0a2ea5cc90.
References
#98624 - Revert "[v2.0.1] `GradScaler` recomputes `found_inf` before `optimizer.step`"
Author
crcrpar
Parents
c8f24700
Loading