amend

facebookresearch · Jul 31, 2024 · e262958 · e262958
1 parent c84a9af
commit e262958
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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_