fixed an issue where the wrong learning rate was overriding the real learning rate for the NN