Update (base update)

[ghstack-poisoned]
pytorch · Sep 2, 2024 · 6b1761e · 6b1761e
1 parent d5a62c4
commit 6b1761e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmarks/test_objectives_benchmarks.py b/benchmarks/test_objectives_benchmarks.py
@@ -320,7 +320,7 @@ def test_sac_speed(
         value, in_keys=["hidden", "action"], out_keys=["state_action_value"]
     )
     value = Seq(common, value_head)
-    value(actor(td))
+    value(actor(td.clone()))
 
     loss = SACLoss(actor, value, action_spec=Unbounded(shape=(n_act,)))
 
@@ -404,7 +404,7 @@ def test_redq_speed(
         value, in_keys=["hidden", "action"], out_keys=["state_action_value"]
     )
     value = Seq(common, value_head)
-    value(actor(td))
+    value(actor(td.copy()))
 
     loss = REDQLoss(actor, value, action_spec=Unbounded(shape=(n_act,)))
 
@@ -489,7 +489,7 @@ def test_redq_deprec_speed(
         value, in_keys=["hidden", "action"], out_keys=["state_action_value"]
     )
     value = Seq(common, value_head)
-    value(actor(td))
+    value(actor(td.copy()))
 
     loss = REDQLoss_deprecated(actor, value, action_spec=Unbounded(shape=(n_act,)))