* handle get_last_lr() before first step() * abstract away the lr getting logic * cleanup * add test * move to utils