Pytorch Print Learning Rate