tests/trainer/test_trainer_utils.py · 47412c7d434f6ddfc02a9b7ecd6182b86ae0a164 · chenpangpang / transformers

Ensure tensors are at least 1d for pad and concat (#17179) · 47412c7d

Antoni Baum authored May 11, 2022

* Ensure tensors are at least 1d for pad and concat

* Compatibility

* Fix

* Fix

* Add test

* Retrigger CI

* Consistency with master

* Retrigger CI

47412c7d

test_trainer_utils.py 21.2 KB

Replace test_trainer_utils.py