I'm getting this error when I run `scheduler = torch.optim.lr_scheduler.LambdaLR(optimizer, lr_lambda)` 