diff --git a/UD_any/launchBatches.py b/UD_any/launchBatches.py
index 40ef2439b2a267ddd4dad796c32309ad04ab6de9..b7f4316f3c9473973b57ce872359d6fc3f010205 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()