fix num_agent bug in MAT_network

OpenRL-Lab · Sep 21, 2023 · 38a9b86 · 38a9b86
1 parent 8797b09
commit 38a9b86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openrl/modules/networks/MAT_network.py b/openrl/modules/networks/MAT_network.py
@@ -420,7 +420,7 @@ def get_actions(
     ):
         obs = obs.reshape(-1, self.n_agent, self.obs_dim)
         if action_masks is not None:
-            action_masks = action_masks.reshape(-1, self.num_agents, self.action_dim)
+            action_masks = action_masks.reshape(-1, self.n_agent, self.action_dim)
 
         # state unused
         ori_shape = np.shape(obs)