accelerate

fixing remaining lr scheduler issues for pytorch nightly

#884

Merged