PR #21991 Fix EMA updates after skipped AMP optimizer steps

Fix EMA updates after skipped AMP optimizer steps #21991

anishhongekar47-dev wants to merge 2 commits into Lightning-AI:master from anishhongekar47-dev:fix/ema-gradscaler-skip

Fix EMA updates after skipped AMP optimizer steps

524d9973

anishhongekar47-dev requested a review from

tchaton 11 days ago

anishhongekar47-dev requested a review from

justusschock 11 days ago

anishhongekar47-dev requested a review from

ethanwharris 11 days ago

fix(callbacks): guard EMA weight averaging against skipped AMP steps

913d979e

anishhongekar47-dev requested a review from

williamFalcon 11 days ago

Reviewers

tchaton

justusschock

ethanwharris

williamFalcon

Assignees

No one assigned

Labels

None yet

Milestone

No milestone