Skip to content

Commit e124cd8

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent cc5407a commit e124cd8

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

tiatoolbox/models/engine/semantic_segmentor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@
1313
import joblib
1414
import numpy as np
1515
import torch
16+
import torch.distributed as dist
1617
import torch.multiprocessing as torch_mp
1718
import torch.utils.data as torch_data
18-
import torch.distributed as dist
1919
import tqdm
2020

2121
from tiatoolbox import logger, rcParam

tiatoolbox/models/models_abc.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,7 @@ def model_to(model: torch.nn.Module, device: str = "cpu") -> torch.nn.Module:
6363
# This assumes a single-process DDP setup for inference
6464
os.environ["MASTER_ADDR"] = "localhost"
6565
os.environ["MASTER_PORT"] = "12355"
66-
dist.init_process_group(
67-
backend="nccl", rank=0, world_size=1
68-
)
66+
dist.init_process_group(backend="nccl", rank=0, world_size=1)
6967
model = DistributedDataParallel(model, device_ids=[device.index])
7068

7169
return model

0 commit comments

Comments
 (0)