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

Update ddp_spawn.py

This commit is contained in:
shuyingsunshine21 2021-03-24 01:37:52 -07:00 committed by GitHub
parent 536c1323b0
commit f17210183b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/plugins/training_type/ddp_spawn.py
View File

@ -21,7 +21,6 @@ import torch.distributed as torch_distrib
import torch.multiprocessing as mp
from torch.nn.parallel.distributed import DistributedDataParallel
from torch.optim import Optimizer
import numpy
from pytorch_lightning.distributed.dist import LightningDistributed
from pytorch_lightning.overrides import LightningDistributedModule
@ -79,7 +78,6 @@ class DDPSpawnPlugin(ParallelPlugin):
def setup(self, model):
os.environ["MASTER_PORT"] = str(self.cluster_environment.master_port())
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
# pass in a state q
smp = mp.get_context("spawn")