Commit Graph

2 Commits (dca98937f834f5af2730f481bf6f5e5eee844742)

Author SHA1 Message Date
Frank Lee 9fdebadd69
[doc] improved docstring in the amp module (#857) 2022-04-25 13:42:17 +08:00
Frank Lee 3d5d64bd10 refactored grad scaler (#338) 2022-03-11 15:50:28 +08:00