DeepSpeed
[BUGFIX] Reset `bucket.elements` after reduction in ZeRO Stage 3
#7418

Merged

[BUGFIX] Reset `bucket.elements` after reduction in ZeRO Stage 3 #7418

tohtana merged 2 commits into deepspeedai:master from rahul713rk:issue7415

reset the `bucket.elements` counter

e3c440d0

rahul713rk requested a review from

tjruwase 81 days ago

rahul713rk requested a review from

tohtana 81 days ago

Merge branch 'master' into issue7415

f05efda0

tohtana approved these changes on 2025-07-08

tohtana enabled auto-merge (squash) 81 days ago

tohtana merged 27228462 into master 81 days ago

Reviewers

tohtana

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone