DeepSpeed
691458f8 - zero_to_fp32.py: Handle a case where shape doesn't have numel attr (#4842)

Commit
1 year ago
zero_to_fp32.py: Handle a case where shape doesn't have numel attr (#4842) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Author
Parents
Loading