diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 57855128..758de820 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -3,6 +3,7 @@ import sys import warnings import logging + import torch import torch.distributed as dist import torch.multiprocessing as mp