src/transformers/utils/generic.py · 3951b9f3908bfa30be7fd814cd2ad1039d3162d8 · chenpangpang / transformers

Add utility to find model labels (#16526) · 3951b9f3

Sylvain Gugger authored Apr 04, 2022



* Add utility to find model labels

* Use it in the Trainer

* Update src/transformers/utils/generic.py
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

* Quality
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

3951b9f3

generic.py 9.59 KB

Replace generic.py