amend

pytorch · Jan 31, 2024 · bf4a0d9 · bf4a0d9
1 parent bd498ab
commit bf4a0d9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/multiagent/iql.py b/examples/multiagent/iql.py
@@ -176,7 +176,7 @@ def train(cfg: "DictConfig"):  # noqa: F821
                 optim.zero_grad()
                 target_net_updater.step()
 
-        qnet_explore.step(frames=current_frames)  # Update exploration annealing
+        qnet_explore[1].step(frames=current_frames)  # Update exploration annealing
         collector.update_policy_weights_()
 
         training_time = time.time() - training_start

diff --git a/examples/multiagent/qmix_vdn.py b/examples/multiagent/qmix_vdn.py
@@ -211,7 +211,7 @@ def train(cfg: "DictConfig"):  # noqa: F821
                 optim.zero_grad()
                 target_net_updater.step()
 
-        qnet_explore.step(frames=current_frames)  # Update exploration annealing
+        qnet_explore[1].step(frames=current_frames)  # Update exploration annealing
         collector.update_policy_weights_()
 
         training_time = time.time() - training_start