accelerate
b626ef5f
- Select the DeepSpeedCPUOptimizer based on the original optimizer class. (#3255)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Select the DeepSpeedCPUOptimizer based on the original optimizer class. (#3255) * Select the DeepSpeedCPUOptimizer based on the original optimizer class. * abstract out optimizer selection to a deepspeed util * add deepspeed cpu Adam & AdamW
References
#3255 - Select the DeepSpeedCPUOptimizer based on the original optimizer class.
Author
eljandoubi
Parents
dd68af88
Loading