PR #31095 Add on_optimizer_step to callback options

Add on_optimizer_step to callback options #31095

LysandreJik merged 4 commits into huggingface:main from dhruvbpai:before_optimizer_step

Modified test

a69f0e93

Added on_optimizer_step to callbacks

95a45795

Move callback after step is called

5dcca4c8

Added on optimizer step callback

87e740ff

younesbelkada approved these changes on 2024-05-29

younesbelkada requested a review from

muellerzr 2 years ago

muellerzr approved these changes on 2024-05-29

muellerzr requested a review from

amyeroberts 2 years ago

LysandreJik approved these changes on 2024-05-29

LysandreJik merged 5c882535 into main 2 years ago

dhruvbpai deleted the before_optimizer_step branch 2 years ago

Reviewers

LysandreJik

muellerzr

younesbelkada

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone