Fix: Cannot convert a MPS Tensor to float64 dtype as the MPS framewor…

…k doesn't support float64.
thu-ml · Jan 19, 2025 · c04cd1b · c04cd1b
1 parent 151000b
commit c04cd1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tianshou/data/utils/converter.py b/tianshou/data/utils/converter.py
@@ -45,10 +45,10 @@ def to_torch(
         x.dtype.type,
         np.bool_ | np.number,
     ):  # most often case
-        x = torch.from_numpy(x).to(device)
+        x = torch.from_numpy(x)
         if dtype is not None:
             x = x.type(dtype)
-        return x
+        return x.to(device)
     if isinstance(x, torch.Tensor):  # second often case
         if dtype is not None:
             x = x.type(dtype)