hassony2 / manopth

MANO layer for PyTorch, generating hand meshes as a differentiable layer
GNU General Public License v3.0
601 stars 104 forks source link

running on GPU #29

Open Pyf1234 opened 1 year ago

Pyf1234 commented 1 year ago

Hello, I found that the tensor in manolayer is not moved to "cuda". I guess it couldn't run on GPU. Is there any implementation which could run on GPU ? Thanks a lot!