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