beta=0.1 learning_rate=5e-5