Dhruv Pai 5c88253556 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

2024-05-29 16:20:59 +02:00

16 KiB

Raw Blame History

View Raw