Uncertainty Weighted Gradients for Model Calibration | Read Paper on Bytez