From dac3456eba5516a23d3ee52d666971cf61fda610 Mon Sep 17 00:00:00 2001
From: yzhilong <yzlong99@gmail.com>
Date: Sat, 24 Feb 2024 13:51:56 +0800
Subject: [PATCH] Add slurm scripts

---
 run_sbatches.sh |  9 +++++++++
 slurm.sh        | 22 +++++++++++++---------
 2 files changed, 22 insertions(+), 9 deletions(-)
 create mode 100644 run_sbatches.sh

diff --git a/run_sbatches.sh b/run_sbatches.sh
new file mode 100644
index 00000000..5e3c9f48
--- /dev/null
+++ b/run_sbatches.sh
@@ -0,0 +1,9 @@
+#!/bin/bash
+
+msg_shapes=(16 32 64)
+algos=('coma' 'maa2c' 'maddpg' 'mappo' 'qmix')
+for msg_shape in "${msg_shapes[@]}"; do
+    for algo in "${algos[@]}"; do
+        sbatch slurm.sh $algo $msg_shape
+    done
+done
\ No newline at end of file
diff --git a/slurm.sh b/slurm.sh
index 336360ec..d42d724a 100644
--- a/slurm.sh
+++ b/slurm.sh
@@ -1,15 +1,19 @@
 #!/bin/bash
 
 #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 --partition=long
+#SBATCH --cpus-per-task=6
+#SBATCH --gpus=1
+#SBATCH --mem=65G
+#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
\ No newline at end of file