Merge pull request #118 from FLAIROx/mappo_rnn_small_fix

small fix for #117
FLAIROx · Oct 28, 2024 · 18a1027 · 18a1027
2 parents 42e7d63 + 05b54c4
commit 18a1027
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/MAPPO/mappo_rnn_smax.py b/baselines/MAPPO/mappo_rnn_smax.py
@@ -255,7 +255,7 @@ def train(rng):
             tx=actor_tx,
         )
         critic_train_state = TrainState.create(
-            apply_fn=actor_network.apply,
+            apply_fn=critic_network.apply,
             params=critic_network_params,
             tx=critic_tx,
         )