Skip to content

Commit 38a9b86

Browse files
committed
fix num_agent bug in MAT_network
1 parent 8797b09 commit 38a9b86

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

openrl/modules/networks/MAT_network.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ def get_actions(
420420
):
421421
obs = obs.reshape(-1, self.n_agent, self.obs_dim)
422422
if action_masks is not None:
423-
action_masks = action_masks.reshape(-1, self.num_agents, self.action_dim)
423+
action_masks = action_masks.reshape(-1, self.n_agent, self.action_dim)
424424

425425
# state unused
426426
ori_shape = np.shape(obs)

0 commit comments

Comments
 (0)