Skip to content

Fix index out of range error in ZeRO optimizer#202

Merged
tjruwase merged 2 commits intomasterfrom
olruwase/zero_index_out_of_range
Apr 24, 2020
Merged

Fix index out of range error in ZeRO optimizer#202
tjruwase merged 2 commits intomasterfrom
olruwase/zero_index_out_of_range

Conversation

@tjruwase
Copy link
Contributor

Fix index out of range error when parameter count is not multiple of ranks.
Closing #129

@ShadenSmith
Copy link
Contributor

Changes look fine to me. If there's a feasible way to add a reproducing unit test, I think that is a good idea too.

@tjruwase
Copy link
Contributor Author

Thanks for the suggestion. I will work on that.

@ShadenSmith
Copy link
Contributor

Awesome, thanks Tunji!

@tjruwase tjruwase merged commit 512a0d4 into master Apr 24, 2020
@jeffra jeffra deleted the olruwase/zero_index_out_of_range branch September 24, 2021 04:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants