tests/test_modeling_tf_common.py · 96d833b211a35bc48c3f9174042a796bb110a66b · chenpangpang / transformers

Return scalar losses instead of per-sample means (#18013) · 96d833b2

Matt authored Jul 04, 2022

* Return scalar losses instead of per-sample means

* Make loss shape (1,) instead of scalar

* Allow scalar losses in test_loss_computation

* Allow scalar losses in test_loss_computation

* Allow scalar losses in test_loss_computation

* Remove XLA loss function for RAG

96d833b2

test_modeling_tf_common.py 100 KB

Replace test_modeling_tf_common.py