tests/test_data_collator.py · f00f22a3e290fd377b979124dcf9800b3d73eb11 · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Fixes tf_default_data_collator sometimes guessing the wrong dtype for labels (#15234) · f00f22a3
Matt authored Jan 20, 2022
```
* Fixes tf_default_data_collator sometimes guessing the wrong dtype for labels

* Add test for numpy scalar inputs
```
f00f22a3

test_data_collator.py 41 KB