DeepSpeed
ZeRO3: Gradient norm allreduce for DP
#1021
Merged

ZeRO3: Gradient norm allreduce for DP #1021

tjruwase merged 2 commits into master from olruwase/zero3_dp_norm_allreduce
tjruwase
tjruwase Handle Norm allreduce when no mp
a9879b89
tjruwase tjruwase requested a review from ShadenSmith ShadenSmith 4 years ago
tjruwase tjruwase requested a review from jeffra jeffra 4 years ago
tjruwase tjruwase requested a review from samyam samyam 4 years ago
tjruwase tjruwase requested a review from arashashari arashashari 4 years ago
tjruwase tjruwase requested a review from awan-10 awan-10 4 years ago
tjruwase tjruwase requested a review from cli99 cli99 4 years ago
tjruwase tjruwase requested a review from conglongli conglongli 4 years ago
tjruwase tjruwase requested a review from eltonzheng eltonzheng 4 years ago
tjruwase tjruwase requested a review from minjiaz minjiaz 4 years ago
tjruwase tjruwase requested a review from niumanar niumanar 4 years ago
tjruwase tjruwase requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 4 years ago
jeffra Merge branch 'master' into olruwase/zero3_dp_norm_allreduce
3e11f4c7
jeffra
jeffra approved these changes on 2021-04-29
tjruwase tjruwase merged 429dfa6c into master 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone