Adaptive Gradient Parameter Updater - Coding problem | OneNoughtOne