From e2e558f1687062f7f51543c5693c1c05d96a5b25 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Mon, 6 Jul 2026 11:00:32 +0200 Subject: [PATCH] higher ram size --- main_distributed.py | 2 +- main_distributed_supervised.py | 2 +- tools/run_grid.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/main_distributed.py b/main_distributed.py index f3ef79c..9a67a98 100644 --- a/main_distributed.py +++ b/main_distributed.py @@ -77,7 +77,7 @@ def launch(): max_num_timeout=20) executor.update_parameters( partition=args.partition, - mem_per_gpu='55G', + mem_per_gpu='256G', time=args.time, nodes=args.nodes, ntasks_per_node=args.tasks_per_node, diff --git a/main_distributed_supervised.py b/main_distributed_supervised.py index 37f2c5e..58a6d48 100644 --- a/main_distributed_supervised.py +++ b/main_distributed_supervised.py @@ -77,7 +77,7 @@ def launch(): ) executor.update_parameters( partition=args.partition, - mem_per_gpu="55G", + mem_per_gpu="256G", time=args.time, nodes=args.nodes, ntasks_per_node=args.tasks_per_node, diff --git a/tools/run_grid.py b/tools/run_grid.py index 1f42a42..656a820 100644 --- a/tools/run_grid.py +++ b/tools/run_grid.py @@ -88,7 +88,7 @@ def main(): ) executor.update_parameters( partition=args.partition or launch.get("partition"), - mem_per_gpu=launch.get("mem_per_gpu", "55G"), + mem_per_gpu=launch.get("mem_per_gpu", "256G"), time=args.time or int(launch.get("time", 4300)), nodes=args.nodes or int(launch.get("nodes", 1)), ntasks_per_node=args.tasks_per_node or int(launch.get("tasks_per_node", 1)),