DeepSpeed
08c96a1b
- ZeRO-1 tune max-elems + bug fix (#532)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
ZeRO-1 tune max-elems + bug fix (#532) * zero-1 memory fix * auto-tune max elems per comm to reduce padding/comm intervals * clean-up and added previously missing reduction options * fix testing backing to work with torch1.7
References
#532 - ZeRO-1 tune max-elems + bug fix
Author
jeffra
Parents
fdd81c30
Loading