From 9ba06819780f97cf8fa6f3d6149e6c75cb4abbaa Mon Sep 17 00:00:00 2001
From: xuetaowave <cxt@cxt-win>
Date: Sun, 18 Feb 2024 20:21:15 +0800
Subject: [PATCH] update
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index d3fb74d..47e33cc 100644
--- a/train.py
+++ b/train.py
@@ -559,7 +559,7 @@ if __name__ == '__main__':
local_rank = 0
if params['world_size'] > 1:
dist.init_process_group(backend='nccl',
- init_method='env://', timeout=timedelta(seconds=10))
+ init_method='env://', timeout=timedelta(seconds=120))
local_rank = int(os.environ["LOCAL_RANK"])
args.gpu = local_rank
world_rank = dist.get_rank()
--
GitLab