onnxruntime
Fix loss scaling when running ORTTrainer with BERT under mixed-precision mode
#6932
Merged

Fix loss scaling when running ORTTrainer with BERT under mixed-precision mode #6932

wschin merged 2 commits into master from wechi/fix-loss-scale-master
wschin
wschin Fix missed Loss scale
58e63dc8
wschin not to dump
e22f94a7
wschin wschin requested a review 5 years ago
wschin wschin added training
wschin wschin added type:bug
SherlockNoMad
SherlockNoMad approved these changes on 2021-03-08
wschin wschin merged de6e66f3 into master 5 years ago
wschin wschin deleted the wechi/fix-loss-scale-master branch 5 years ago
wschin

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone