Global Norm Gradient Rescaling - Coding problem | OneNoughtOne