accelerate
Select the DeepSpeedCPUOptimizer based on the original optimizer class.
#3255
Merged
Merged
Select the DeepSpeedCPUOptimizer based on the original optimizer class. #3255
muellerzr
merged
4 commits into
huggingface:main
from
eljandoubi:add_more_cpu_optimizers
Select the DeepSpeedCPUOptimizer based on the original optimizer class.
8bb69927
muellerzr
commented
on 2024-11-25
Merge branch 'huggingface:main' into add_more_cpu_optimizers
92a2e1ff
abstract out optimizer selection to a deepspeed util
1cdbb407
add deepspeed cpu Adam & AdamW
53e5400f
muellerzr
approved these changes
on 2024-12-02
Login to write a write a comment.
Login via GitHub