Update

[ghstack-poisoned]
pytorch · Dec 16, 2024 · 3cff702 · 3cff702
1 parent 7cba108
commit 3cff702
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchrl/data/tensor_specs.py b/torchrl/data/tensor_specs.py
@@ -407,12 +407,12 @@ def unbind(self, dim: int = 0):
     @low.setter
     def low(self, value):
         self.device = value.device
-        self._low = value.cpu()
+        self._low = value
 
     @high.setter
     def high(self, value):
         self.device = value.device
-        self._high = value.cpu()
+        self._high = value
 
     def __post_init__(self):
         self.low = self.low.clone()