Update lr_decay handling in trainera