From 69d811ecb47739724de56a7b0e89a2f9300c1780 Mon Sep 17 00:00:00 2001 From: "ahyanc25@reutlingen-university.de" Date: Sat, 18 Apr 2026 12:28:40 +0200 Subject: [PATCH] slurm version of main_distributed.py --- main_distributed.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/main_distributed.py b/main_distributed.py index 7cb3846..8ec6d38 100644 --- a/main_distributed.py +++ b/main_distributed.py @@ -73,15 +73,15 @@ class Trainer: def launch(): - executor = submitit.AutoExecutor( + executor = submitit.SlurmExecutor( folder=os.path.join(args.folder, 'job_%j'), - slurm_max_num_timeout=20) + max_num_timeout=20) executor.update_parameters( - slurm_partition=args.partition, - slurm_mem_per_gpu='55G', - timeout_min=args.time, + partition=args.partition, + mem_per_gpu='55G', + time=args.time, nodes=args.nodes, - tasks_per_node=args.tasks_per_node, + ntasks_per_node=args.tasks_per_node, cpus_per_task=10, gpus_per_node=args.tasks_per_node)