Adaptive Moment Estimation Optimizer - Coding problem | OneNoughtOne