onnxruntime
Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss.
#4509

Merged

Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss. #4509

codemzs merged 4 commits into master from bert_loss_function

Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss.

1fdac2b5

codemzs requested a review 5 years ago

codemzs added training

Update BERT loss function with correct logit shapes for softmax cross…

3039c5ab

Merge branch 'master' of https://github.com/microsoft/onnxruntime int…

ed0109f0

ytaous commented on 2020-07-15

fix test and PR comments.

563dc098

ytaous approved these changes on 2020-07-16

codemzs merged b43ce2d7 into master 5 years ago

codemzs deleted the bert_loss_function branch 5 years ago

Reviewers

ytaous

Assignees

No one assigned

Labels

training

Milestone

No milestone