PR #18013 Return scalar losses instead of per-sample means

Return scalar losses instead of per-sample means #18013

Rocketknight1 merged 6 commits into main from xla_loss_fixes

Return scalar losses instead of per-sample means

2521c686

Rocketknight1 requested a review from

ydshieh 4 years ago

Rocketknight1 requested a review from

gante 4 years ago

ydshieh approved these changes on 2022-07-04

Make loss shape (1,) instead of scalar

b339284c

Allow scalar losses in test_loss_computation

d6f834ec

Allow scalar losses in test_loss_computation

4918a761

Allow scalar losses in test_loss_computation

633664a0

Remove XLA loss function for RAG

f3515aa9

gante approved these changes on 2022-07-04

Rocketknight1 merged 96d833b2 into main 4 years ago

Rocketknight1 deleted the xla_loss_fixes branch 4 years ago

ydshieh commented on 2022-07-05

Reviewers

ydshieh

gante

Assignees

No one assigned

Labels

None yet

Milestone

No milestone