added config yaml files

nadasimre · nadasimre · commit cdeb12cda0bb · 2025-06-01T20:54:04.000+02:00
diff --git a/marl-task/conf/base.yaml b/marl-task/conf/base.yaml
@@ -0,0 +1,8 @@
+defaults:
+  - model: ppo
+  - env: swarm
+
+frames_per_batch: 4096
+minibatch_size: 256
+clip_epsilon: 0.2
+entropy_coef: 0.01
diff --git a/marl-task/conf/env/swarm.yaml b/marl-task/conf/env/swarm.yaml
@@ -0,0 +1,4 @@
+name: swarm
+landscape_dim: 2
+num_agents: 10
+batch_size: 16
diff --git a/marl-task/conf/model/ppo.yaml b/marl-task/conf/model/ppo.yaml
@@ -0,0 +1,4 @@
+name: ppo
+hidden_sizes: [128, 64]
+activation: relu
+learning_rate: 3e-4
