fix bug; restore learning_rate