tests/trainer/test_trainer_callback.py · 5c88253556b7f15cf7d7e9793d7b2a39b4aa588a · chenpangpang / transformers

Dhruv Pai authored May 29, 2024

* Modified test

* Added on_optimizer_step to callbacks

* Move callback after step is called

* Added on optimizer step callback

5c882535

test_trainer_callback.py 16.4 KB