fix learning rate bug