Giving higher weights to l2 loss doesn’t improve the result. I need to check the gradients of different layer. Could you please give reference code or link to plot and visualize  the gradients of different layer?