fix learning rate and momentum bug in nntrain_gpu