From e0bd9fc7ce32c270b847dfdf1bdae5cee6927976 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Sat, 16 May 2026 12:47:41 +0200 Subject: [PATCH] import dist --- src/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/train.py b/src/train.py index 29f0722..413742e 100644 --- a/src/train.py +++ b/src/train.py @@ -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