Skip to content

ZeRO3: Gradient norm allreduce for DP#1021

Merged
tjruwase merged 2 commits intomasterfrom
olruwase/zero3_dp_norm_allreduce
Apr 30, 2021
Merged

ZeRO3: Gradient norm allreduce for DP#1021
tjruwase merged 2 commits intomasterfrom
olruwase/zero3_dp_norm_allreduce

Conversation

@tjruwase
Copy link
Copy Markdown
Contributor

Compute gradient norm allreduce for DP training. Fix for #966.

@jeffra jeffra linked an issue Apr 29, 2021 that may be closed by this pull request
@tjruwase tjruwase merged commit 429dfa6 into master Apr 30, 2021
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.

ZeRO 0, 1, 2, 3 produce different results

2 participants