* [Feature]: Add parameter schedulers. * update * update * update * update * add docstring to lr and momentum * resolve comments
