From 3e588c616c8efc3ef9a2637dfab0e2aa831873c8 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Mon, 6 Jul 2026 11:10:45 +0200 Subject: [PATCH] reduce ram --- 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 9a67a98..eb44dc5 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='256G', + mem_per_gpu='180G', 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 58a6d48..6f5c6b6 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="256G", + mem_per_gpu="180G", 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 656a820..f4a1317 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", "256G"), + mem_per_gpu=launch.get("mem_per_gpu", "180G"), 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)),