DeepSpeed
Spread layers more uniformly when using partition_uniform
#4053

Merged

Commits

update partition_uniform util function

marcobellagente93 committed 3 years ago
formatting

marcobellagente93 committed 3 years ago
Merge branch 'master' into more-uniform-pipeline

marcobellagente93 committed 3 years ago
Merge branch 'master' into more-uniform-pipeline

tjruwase committed 3 years ago
Merge branch 'master' into more-uniform-pipeline

marcobellagente93 committed 3 years ago
Merge branch 'master' into more-uniform-pipeline

marcobellagente93 committed 3 years ago
Merge branch 'master' into more-uniform-pipeline

marcobellagente93 committed 3 years ago