mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-04 13:00:36 +08:00
fixing Win failed import (#1163)
* version * try fix distrib * update try import
This commit is contained in:
1 parent
49d000c0c9
commit
e461ec0037
8 files changed
+18
-19
No files matched your search
@@ -8,7 +8,7 @@ from typing import Union, Optional, List, Dict, Tuple, Iterable
|
||||
|
||||
import torch
|
||||
from torch import optim
|
||||
import torch.distributed as dist
|
||||
import torch.distributed as torch_distrib
|
||||
import torch.multiprocessing as mp
|
||||
from torch.optim.optimizer import Optimizer
|
||||
from torch.utils.data import DataLoader
|
||||
@@ -748,7 +748,7 @@ class Trainer(
|
||||
self.logger.save()
|
||||
|
||||
if self.use_ddp or self.use_ddp2:
|
||||
dist.barrier()
|
||||
torch_distrib.barrier()
|
||||
|
||||
# wait for all models to restore weights
|
||||
if self.on_tpu and XLA_AVAILABLE:
|
||||
|
||||
Reference in new issue
Block a user