Exponential Decay Learning Rate Scheduler - Coding problem | OneNoughtOne