Do not unscale the gradients if loss scale equal to 1 (#748)
* Do not unscale the gradients if loss scale equal to 1 * Disable unscaling loss scale == 1 only for static scaling
Showing
Please register or sign in to comment
* Do not unscale the gradients if loss scale equal to 1 * Disable unscaling loss scale == 1 only for static scaling