learning rate and momentum bug