diff --git a/offpolicy/algorithms/maddpg/algorithm/actor_critic.py b/offpolicy/algorithms/maddpg/algorithm/actor_critic.py index 8076c97..bb2e304 100644 --- a/offpolicy/algorithms/maddpg/algorithm/actor_critic.py +++ b/offpolicy/algorithms/maddpg/algorithm/actor_critic.py @@ -64,7 +64,7 @@ def __init__(self, args, central_obs_dim, central_act_dim, device, num_q_outs=1) init_method = [nn.init.xavier_uniform_, nn.init.orthogonal_][self._use_orthogonal] def init_(m): return init(m, init_method, lambda x: nn.init.constant_(x, 0)) - self.q_outs = [init_(nn.Linear(self.hidden_size, 1)) for _ in range(num_q_outs)] + self.q_outs = nn.ModuleList([init_(nn.Linear(self.hidden_size, 1)) for _ in range(num_q_outs)]) self.to(device)