From 05050d07762cf9681221cbe20ac080b20dba4c38 Mon Sep 17 00:00:00 2001
From: alex <amacrutherford@gmail.com>
Date: Wed, 27 Mar 2024 17:32:11 +0000
Subject: [PATCH] mpe sweep

---
 baselines/IPPO/config/ippo_rnn_mpe.yaml |  4 +--
 jaxmarl_mpe_sweep.yaml                  | 34 +++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 2 deletions(-)
 create mode 100644 jaxmarl_mpe_sweep.yaml

diff --git a/baselines/IPPO/config/ippo_rnn_mpe.yaml b/baselines/IPPO/config/ippo_rnn_mpe.yaml
index 0946a937..0b8ffc09 100644
--- a/baselines/IPPO/config/ippo_rnn_mpe.yaml
+++ b/baselines/IPPO/config/ippo_rnn_mpe.yaml
@@ -19,6 +19,6 @@
 "SEED": 30
 
 # WandB Params
-"ENTITY": ""
+"ENTITY": "amacrutherford"
 "PROJECT": "jaxmarl-mpe"
-"WANDB_MODE" : "disabled"
\ No newline at end of file
+"WANDB_MODE" : "online"
\ No newline at end of file
diff --git a/jaxmarl_mpe_sweep.yaml b/jaxmarl_mpe_sweep.yaml
new file mode 100644
index 00000000..05075448
--- /dev/null
+++ b/jaxmarl_mpe_sweep.yaml
@@ -0,0 +1,34 @@
+command:
+  - python3
+  - ${program}
+  - ${args_no_hyphens}
+entity: amacrutherford
+method: grid
+parameters:
+  ENV_NAME:
+    values:
+      - "MPE_simple_spread_v3"
+      - "MPE_simple_tag_v3"
+      - "MPE_simple_world_comm_v3"
+      - "MPE_simple_spread_v3"
+      - "MPE_simple_crypto_v3"
+      - "MPE_simple_speaker_listener_v4"
+      - "MPE_simple_push_v3"
+      - "MPE_simple_adversary_v3"
+      - "MPE_simple_reference_v3"
+
+  SEED:
+    values:
+      - 42
+      - 43
+      - 44
+      - 45
+      - 46
+      # - 47
+      # - 48
+      # - 49
+      # - 50
+      # - 51
+
+program: baselines/IPPO/ippo_rnn_mpe.py
+project: jaxmarl-smax
\ No newline at end of file