DeepSpeed

fix typo and remove duplicated code in ZeRO stage 1 and 2

#2655

Merged