DeepSpeed
DeepSpeed copied to clipboard
zero_to_fp32.py: Handle a case where shape doesn't have numel attr