Add lr_decay