alg(wyh):add madqn policy test codes

opendilab · Nov 15, 2022 · 2ae1a9d · 2ae1a9d
1 parent 35e0071
commit 2ae1a9d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py b/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py
@@ -27,6 +27,7 @@
             obs_shape=2 + 2 + n_landmark * 2 + (n_agent - 1) * 2 + (n_agent - 1) * 2,
             global_obs_shape=2 + 2 + n_landmark * 2 + (n_agent - 1) * 2 + (n_agent - 1) * 2 + n_agent * (2 + 2) +
             n_landmark * 2 + n_agent * (n_agent - 1) * 2,
+            agent_num=n_agent,
             action_shape=5,
             global_cooperation=True,
             hidden_size_list=[256, 256],