Epoch-Based Learning Rate Decay Scheduler - Coding problem | OneNoughtOne