diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 5da1ead938..c21cf3be01 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -10,7 +10,6 @@ import re import torch from torch.utils.data.distributed import DistributedSampler -from torch.optim.lr_scheduler import MultiStepLR import torch.multiprocessing as mp import torch.distributed as dist import numpy as np