DeepSpeed
0e357666 - stage_1_and_2.py: do gradient scale only for fp16 (#3166)

Commit

2 years ago

stage_1_and_2.py: do gradient scale only for fp16 (#3166) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

Author

guoyejun

Parents