[rllib] Add QMIX mixer parameters to optimizer param list (#5014)

* add mixer params * Update qmix_policy.py
2026-06-28 07:18:24 +08:00 · 2019-06-25 19:02:40 -07:00
parent 0131353d42
commit aa5fc52e32
1 changed files with 2 additions and 0 deletions
@@ -204,6 +204,8 @@ class QMixTorchPolicy(Policy):

        # Setup optimizer
        self.params = list(self.model.parameters())
+        if self.mixer:
+            self.params += list(self.mixer.parameters())
        self.loss = QMixLoss(self.model, self.target_model, self.mixer,
                             self.target_mixer, self.n_agents, self.n_actions,
                             self.config["double_q"], self.config["gamma"])