remove lr_scheduler redundancy
This PR solves https://github.com/huggingface/transformers/issues/6374
by removing a hardcoded `lr_scheduler` and switching to using the new method.
@sshleifer has a better version in works, closing.