Decoupled Weight Decay Parameter Update - Coding problem | OneNoughtOne