Matt 96d833b211 Return scalar losses instead of per-sample means (#18013) ...

* 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

2022-07-04 17:26:19 +01:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_tf_xlnet.py

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

2022-07-04 17:26:19 +01:00

test_modeling_xlnet.py

Fx support for multiple model architectures (#17393)

2022-05-31 10:02:55 +02:00

test_tokenization_xlnet.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00