amend

pytorch · Oct 11, 2024 · 3e3a528 · 3e3a528
1 parent 8aac458
commit 3e3a528
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchrl/objectives/utils.py b/torchrl/objectives/utils.py
@@ -257,6 +257,7 @@ def init_(self) -> None:
                 "to True if it is not done by default. "
                 f"If no target parameter is needed, do not use a target updater such as {type(self)}."
             )
+
         # filter the target_ out
         def filter_target(key):
             if isinstance(key, tuple):
@@ -355,7 +356,7 @@ def __init__(
     def _step(
         self, p_source: Tensor | TensorDictBase, p_target: Tensor | TensorDictBase
     ) -> None:
-        p_target.data.mul_(self.eps).add_(p_source.data, (1 - self.eps))
+        p_target.data.lerp_(p_source.data, 1 - self.eps)
 
 
 class HardUpdate(TargetNetUpdater):