onnxruntime
f3b0c93a - Fix issue preventing loss scaler to run due (#4833)

Commit
5 years ago
Fix issue preventing loss scaler to run due (#4833) `LossScaler.update()` was not being properly called due to the incorrect TrainStepInfo.all_finite assignment. Additionally to this fix, _ORTTrainerModelDesc.is_finite was renamed to _ORTTrainerModelDesc.all_finite to make it more uniform with TrainStepInfo
Author
Thiago Crepaldi
Parents
Loading