Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

adjusted imports

This commit is contained in:
William Falcon 2019-07-08 20:11:20 -04:00
parent 7123dfeaf5
commit 7ab1a837a9
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pytorch_lightning/models/trainer.py
View File

@ -1,19 +1,20 @@
import torch
import tqdm
import numpy as np
from pytorch_lightning.root_module.memory import get_gpu_memory_map
from time import sleep
import subprocess
import traceback
from pytorch_lightning.root_module.model_saving import TrainerIO
import warnings
import os
import torch
from torch.utils.data.distributed import DistributedSampler
from torch.optim.lr_scheduler import MultiStepLR
from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel
import pdb
import torch.multiprocessing as mp
import torch.distributed as dist
import os
import subprocess
from time import sleep
from torch.utils.data.distributed import DistributedSampler
import warnings
import numpy as np
import tqdm
from pytorch_lightning.root_module.memory import get_gpu_memory_map
from pytorch_lightning.root_module.model_saving import TrainerIO
from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel
try: