DeepSpeed
b3388e14 - Fix partition id in the fp32->fp16 param copying step for z2+cpu-offload (#2059)

Comment changes are shownComment changes are hidden
Commit
3 years ago
Fix partition id in the fp32->fp16 param copying step for z2+cpu-offload (#2059) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
  • deepspeed/runtime/zero
    • File
      stage_1_and_2.py