From e2629585ad770d5fe456a698082184984b1c10e6 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Wed, 31 Jul 2024 15:42:35 +0200 Subject: [PATCH] amend --- benchmarl/conf/task/vmas/repeat_last.yaml | 2 +- benchmarl/environments/vmas/common.py | 1 + fine_tuned/vmas/conf/config.yaml | 6 +++--- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/benchmarl/conf/task/vmas/repeat_last.yaml b/benchmarl/conf/task/vmas/repeat_last.yaml index 86440fba..b61ac57f 100644 --- a/benchmarl/conf/task/vmas/repeat_last.yaml +++ b/benchmarl/conf/task/vmas/repeat_last.yaml @@ -5,4 +5,4 @@ defaults: max_steps: 100 horizon: 100 n_agents: 4 -k: 1 +k: 4 diff --git a/benchmarl/environments/vmas/common.py b/benchmarl/environments/vmas/common.py index 691594e4..db3ca334 100644 --- a/benchmarl/environments/vmas/common.py +++ b/benchmarl/environments/vmas/common.py @@ -44,6 +44,7 @@ class VmasTask(Task): SIMPLE_SPREAD = None SIMPLE_TAG = None SIMPLE_WORLD_COMM = None + REPEAT_LAST = None def get_env_fun( self, diff --git a/fine_tuned/vmas/conf/config.yaml b/fine_tuned/vmas/conf/config.yaml index 7dcf246e..e9d7afc7 100644 --- a/fine_tuned/vmas/conf/config.yaml +++ b/fine_tuned/vmas/conf/config.yaml @@ -1,8 +1,8 @@ defaults: - experiment: base_experiment - - algorithm: ??? - - task: ??? - - model: layers/mlp + - algorithm: ippo + - task: vmas/repeat_last + - model: layers/gru - model@critic_model: layers/mlp - _self_