Change learning rate when doing SGD