transformers/tests/models/xlnet
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
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
test_tokenization_xlnet.py