src/transformers/trainer_pt_utils.py · 855b95ce345b8eb8295fb969d36f932adada12a5 · chenpangpang / transformers

Safe import of LRScheduler (#29919) · 855b95ce

amyeroberts authored Mar 28, 2024



* Safe import of LRScheduler

* Update src/transformers/trainer_pt_utils.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* Update src/transformers/trainer_pt_utils.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* Fix up

---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

855b95ce

trainer_pt_utils.py 53.6 KB

Replace trainer_pt_utils.py