Skip to content

Commit 86a3168

Browse files
committed
Fix default device issue
1 parent e0e1131 commit 86a3168

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pytorch_kinematics/urdf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def _convert_transform(origin):
1414
if origin is None:
1515
return tf.Transform3d()
1616
else:
17-
rpy = torch.tensor(origin.rpy, dtype=torch.float32)
17+
rpy = torch.tensor(origin.rpy, dtype=torch.float32, device="cpu")
1818
return tf.Transform3d(rot=tf.quaternion_from_euler(rpy, "sxyz"), pos=origin.xyz)
1919

2020

0 commit comments

Comments
 (0)