Skip to content
Snippets Groups Projects
Commit 96e5f780 authored by Franck Dary's avatar Franck Dary
Browse files

Added qos in slurm files

parent dc122663
No related branches found
No related tags found
No related merge requests found
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment