import dist

2026-05-16 12:47:41 +02:00
parent 79fc9564ea
commit e0bd9fc7ce
1 changed files with 1 additions and 0 deletions
@@ -25,6 +25,7 @@ import yaml
 import numpy as np

 import torch
+import torch.distributed as dist
 import torch.multiprocessing as mp
 import torch.nn.functional as F
 from torch.nn.parallel import DistributedDataParallel