diff --git a/execute_vehicle_classification.sh b/execute_vehicle_classification.sh
index 1af31d9c5b2bdf442adb896c167c81793d9fbc30..e8846b789fdf647f0d4e594f200eb13da3a7c94d 100644
--- a/execute_vehicle_classification.sh
+++ b/execute_vehicle_classification.sh
@@ -1,9 +1,9 @@
 #!/bin/bash
+#SBATCH --partition=gpu_4
+#SBATCH --gres=gpu:4
 #SBATCH --ntasks=1
-#SBATCH --time=10
-#SBATCH --mem=5000mb
+#SBATCH --time=20
 #SBATCH --job-name=vehicle_classification
-#SBATCH --partition=dev_single
 
 echo "creating folder"
 #create folder and change directory