diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 203a589a..12c501db 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -1,12 +1,12 @@ """ The trainer handles all the logic for running a val loop, training loop, distributing, etc... """ -from time import sleep import subprocess import traceback import warnings import os import pdb +import re import torch from torch.utils.data.distributed import DistributedSampler