diff --git a/execute_vehicle_classification.sh b/execute_vehicle_classification.sh
index e841c99bd15b24f70ad0477d42e0c6a28da19454..1d363b5699cfc2a7c4210ad7213774b8d462fd76 100644
--- a/execute_vehicle_classification.sh
+++ b/execute_vehicle_classification.sh
@@ -1,8 +1,10 @@
 #!/bin/bash
-#SBATCH --partition=single
-#SBATCH --gres=gpu:4
-#SBATCH --ntasks=1
 #SBATCH --time=20
+#SBATCH --mem=16000
+#SBATCH --ntasks=1
+#SBATCH --partition=gpu_8
+#SBATCH --mem-per-cpu=94000
+#SBATCH --gres=gpu:4
 #SBATCH --job-name=vehicle_classification
 
 echo "creating folder"