Update mujoco_sac.py

thu-ml · Oct 11, 2023 · bb8f814 · bb8f814
1 parent 66b7fc5
commit bb8f814
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mujoco/mujoco_sac.py b/examples/mujoco/mujoco_sac.py
@@ -120,8 +120,8 @@ def test_sac(args=get_args()):
     policy = SACPolicy(
         actor=actor,
         actor_optim=actor_optim,
-        critic=critic1,
-        critic_optim=critic1_optim,
+        critic1=critic1,
+        critic1_optim=critic1_optim,
         critic2=critic2,
         critic2_optim=critic2_optim,
         tau=args.tau,