onnxruntime
b43ce2d7 - Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss. (#4509)

Commit
5 years ago
Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss. (#4509) * Replace loss function in BERT_LOSS with SoftmaxCrossEntropyLoss. * Update BERT loss function with correct logit shapes for softmax cross entropy loss. * fix test and PR comments.
Author
Parents
Loading