accelerate
727d6243 - Add support for deepspeed optimizer and custom scheduler (#1909)

Commit
2 years ago
Add support for deepspeed optimizer and custom scheduler (#1909) * support for deepspeed optimizer and custom scheduler * don't throw the error * Add tests * fix the tests * fix the code quality * Update tests/deepspeed/test_deepspeed.py Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com> * fix the docstrings --------- Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
Author
Parents
Loading