From 252ca3b3ab080eba20a06b286ef0eb5c9bd9bd56 Mon Sep 17 00:00:00 2001
From: Etienne Guevel <etienneguev@gmail.com>
Date: Wed, 15 May 2024 16:52:13 +0200
Subject: [PATCH] fix mem_per_gpu

---
 dinov2/utils/cluster.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dinov2/utils/cluster.py b/dinov2/utils/cluster.py
index 3736bd6..ba0a14b 100644
--- a/dinov2/utils/cluster.py
+++ b/dinov2/utils/cluster.py
@@ -76,8 +76,8 @@ def get_slurm_executor_parameters(
 ) -> Dict[str, Any]:
     # create default parameters
     params = {
-        "mem_gb": 0,  # Requests all memory on a node, see https://slurm.schedmd.com/sbatch.html
-        "mem_per_gb": mem_per_gpu,
+        #"mem_gb": 0,  # Requests all memory on a node, see https://slurm.schedmd.com/sbatch.html
+        "slurm_mem_per_gpu": mem_per_gpu,
         "gpus_per_node": num_gpus_per_node,
         "tasks_per_node": num_gpus_per_node,  # one task per GPU
         "cpus_per_task": 10,