Augment the learning rate