update

allenai · Sep 11, 2024 · 1529d89 · 1529d89
1 parent 8e03caa
commit 1529d89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/allenact/algorithms/onpolicy_sync/engine.py b/allenact/algorithms/onpolicy_sync/engine.py
@@ -1150,7 +1150,7 @@ def single_batch_generator(streaming_storage: StreamingStorageMixin):
 
                 if batch_idx == training_settings.num_mini_batch - 1:
                     if hasattr(self.actor_critic, "update_networks"):
-                        self.actor_critic.update_networks(info)
+                        self.actor_critic.update_networks(**info)
 
     def close(self, verbose=True):
         self._is_closing = True