a tiny bit of optimization in the training.