Update slurm scripts

edu-ai · Mar 1, 2024 · 964ef0b · 964ef0b
1 parent eb535a5
commit 964ef0b
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 9 deletions.
diff --git a/slurm.sh b/slurm.sh
@@ -2,14 +2,18 @@
 
 #SBATCH --job-name=epymarl
 #SBATCH --partition=long                        
-#SBATCH --cpus-per-task=2
-#SBATCH --gres=gpu:a100:1
-#SBATCH --mem=65G                                     
-#SBATCH --time=24:00:00
+#SBATCH --cpus-per-task=6
+#SBATCH --gpus=1
+#SBATCH --mem=80G                                     
+#SBATCH --time=120:00:00
 
 config=$1
-env_name=${2:-"pressureplate:pressureplate-linear-4p-v0"}
-env_time=${3:-25}
-env_config=${4:-gymma}
+message_shape=${2:-32}
+trajectory_length=${3:-3}
+env_name=${4:-MMM}
+env_config=${5:-sc2}
 
-python3 src/main.py --config=$config --env-config=$env_config with env_args.time_limit=$env_time env_args.key=$env_name
+cd /home/y/yzhilong/epymarl
+source ../python_envs/epymarl38/bin/activate
+echo "message_shape: "$message_shape
+python3 src/main.py --config=$config --env-config=$env_config with env_args.map_name=$env_name message_shape=$message_shape trajectory_length=$trajectory_length
diff --git a/slurm_a100.sh b/slurm_a100.sh
@@ -3,7 +3,7 @@
 #SBATCH --job-name=epymarl
 #SBATCH --partition=long                        
 #SBATCH --cpus-per-task=6
-#SBATCH --gres=gpu:a100:1
+#SBATCH --gres=gpu:a100mig:1
 #SBATCH --mem=65G                                     
 #SBATCH --time=120:00:00
 

diff --git a/slurm_no_comms.sh b/slurm_no_comms.sh
@@ -0,0 +1,17 @@
+#!/bin/bash
+
+#SBATCH --job-name=epymarl
+#SBATCH --partition=long                        
+#SBATCH --cpus-per-task=6
+#SBATCH --gpus=1
+#SBATCH --mem=80G                                     
+#SBATCH --time=120:00:00
+
+config=$1
+env_name=${2:-MMM}
+env_config=${3:-sc2}
+
+cd /home/y/yzhilong/epymarl
+source ../python_envs/epymarl38/bin/activate
+echo "message_shape: "$message_shape
+python3 src/main.py --config=$config --env-config=$env_config with env_args.map_name=$env_name allow_communications=false