From 96e5f780bb103841b383bd9f2b2be8af26072a2c Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 1 Jun 2020 12:22:02 +0200
Subject: [PATCH] Added qos in slurm files
---
UD_any/launchBatches.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/UD_any/launchBatches.py b/UD_any/launchBatches.py
index 40ef243..b7f4316 100755
--- a/UD_any/launchBatches.py
+++ b/UD_any/launchBatches.py
@@ -73,6 +73,7 @@ def launchTrainSlurm(mode, expName, arguments, nbHours) :
#SBATCH --gres=gpu:1
#SBATCH --hint=nomultithread
#SBATCH --partition=gpu_p1
+#SBATCH --qos={}
#SBATCH --time={}:00:00
module purge
@@ -80,7 +81,7 @@ module load gcc/9.1.0
module load python/3.7.5
./train.sh {} bin/{} {} --silent
-""".format(expName, expName, expName, nbHours, mode, expName, arguments), file=sFile)
+""".format(expName, expName, expName, "qos_gpu-t4" if nbHours > 20 else "qos_gpu-t3", nbHours, mode, expName, arguments), file=sFile)
sFile.close()
subprocess.Popen("sbatch {}".format(filename), shell=True).wait()
--
GitLab