DeepSpeed
a3be0d44
- TestEmptyParameterGroup: replace fusedAdam with torch.optim.AdamW (#5139)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
TestEmptyParameterGroup: replace fusedAdam with torch.optim.AdamW (#5139) to avoid cuda specific optimizer, with general one. Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
References
#5139 - TestEmptyParameterGroup: replace fusedAdam with torch.optim.AdamW
Author
nelyahu
Parents
9bd62e0b
Loading