mqmix hypernet b2

[zcyyyyyyyyyyy]

in mqmix mixer

self.hyper_b2 = nn.Sequential(
init_(nn.Linear(self.cent_obs_dim, self.hypernet_hidden_dim)),
nn.ReLU(),
init_(nn.Linear(self.hypernet_hidden_dim, 1))
).to(self.device)

should be

self.hyper_b2 = nn.Sequential(
init_(nn.Linear(self.cent_obs_dim, self.mixer_hidden_dim)),
nn.ReLU(),
init_(nn.Linear(self.mixer_hidden_dim, 1))
).to(self.device)

?