alg(wyh):add madqn policy test codes

opendilab · Nov 15, 2022 · 35e0071 · 35e0071
1 parent 88c825c
commit 35e0071
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py b/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py
@@ -24,7 +24,7 @@
         cuda=True,
         nstep=3,
         model=dict(
-            agent_obs_shape=2 + 2 + n_landmark * 2 + (n_agent - 1) * 2 + (n_agent - 1) * 2,
+            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,
             action_shape=5,