wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixing Win failed import (#1163)

Browse Source 
* version

* try fix distrib

* update try import
This commit is contained in:
Jirka Borovec
2020-03-16 20:50:36 -04:00
committed by GitHub
parent 49d000c0c9
commit e461ec0037
8 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/training_io.py
+2 -2
View File
@@ -100,7 +100,7 @@ from subprocess import call
from typing import Union
import torch
import torch.distributed as dist
import torch.distributed as torch_distrib
from pytorch_lightning.core.lightning import LightningModule
from pytorch_lightning.loggers import LightningLoggerBase
@@ -177,7 +177,7 @@ class TrainerIOMixin(ABC):
# wait for all models to restore weights
if self.use_ddp or self.use_ddp2:
# wait for all processes to catch up
dist.barrier()
torch_distrib.barrier()
# wait for all models to restore weights
if self.on_tpu and XLA_AVAILABLE: