training. Return: `torch.optim.lr_scheduler.LambdaLR` with the appropriate schedule. ©