transformers

5c882535 - Add on_optimizer_step to callback options (#31095)

Add on_optimizer_step to callback options (#31095) * Modified test * Added on_optimizer_step to callbacks * Move callback after step is called * Added on optimizer step callback