resolved the bug to get fingertips

otaheri · May 3, 2021 · 4b3e48e · 4b3e48e
1 parent 8ca8aa8
commit 4b3e48e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mano/model.py b/mano/model.py
@@ -301,9 +301,10 @@ def extra_repr(self):
 
     def add_joints(self,vertices,joints, joint_ids = None):
 
+        dev = vertices.device
         if joint_ids is None:
             joint_ids = to_tensor(list(self.tip_ids.values()),
-                                  dtype=torch.long)
+                                  dtype=torch.long).to(dev)
         extra_joints = torch.index_select(vertices, 1, joint_ids)
         joints = torch.cat([joints, extra_joints], dim=1)