PyTorchGradScaler
: Cache _found_inf
on the CPU
#746
Merged
PyTorchGradScaler
: Cache _found_inf
on the CPU
#746