DeepSpeed
a3be0d44 - TestEmptyParameterGroup: replace fusedAdam with torch.optim.AdamW (#5139)

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>
Author
Parents
Loading