diff --git a/aimodel/slurm-pretrain-plot.job b/aimodel/slurm-pretrain-plot.job
index 197f33b..c6d6da7 100755
--- a/aimodel/slurm-pretrain-plot.job
+++ b/aimodel/slurm-pretrain-plot.job
@@ -4,7 +4,7 @@
 #SBATCH -n 14
 #SBATCH -o %j.%N.%a.out.log
 #SBATCH -e %j.%N.%a.err.log
-#SBATCH -p compute
+#SBATCH -p highmem
 #SBATCH --time=3-00:00:00
 #SBATCH --exclusive