From 06ab751b2f43fae8ff8e9dd73a87c4638aae73d1 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 11 Jul 2023 13:29:51 +0000 Subject: [PATCH] uuguuguguguuuu --- references/detection/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/references/detection/utils.py b/references/detection/utils.py index f67226ffedb..f73915580f7 100644 --- a/references/detection/utils.py +++ b/references/detection/utils.py @@ -262,9 +262,9 @@ def init_distributed_mode(args): args.rank = int(os.environ["RANK"]) args.world_size = int(os.environ["WORLD_SIZE"]) args.gpu = int(os.environ["LOCAL_RANK"]) - # elif "SLURM_PROCID" in os.environ: - # args.rank = int(os.environ["SLURM_PROCID"]) - # args.gpu = args.rank % torch.cuda.device_count() + elif "SLURM_PROCID" in os.environ: + args.rank = int(os.environ["SLURM_PROCID"]) + args.gpu = args.rank % torch.cuda.device_count() else: print("Not using distributed mode") args.distributed = False