Python Learning Rate Decay