Error fix for "Expected all tensors to be on the same device"

rotem-shalev / Ham2Pose

Official implementation for "Ham2Pose: Animating Sign Language Notation into Pose Sequences" [CVPR 2023]

https://rotem-shalev.github.io/ham-to-pose

43 stars 5 forks source link

Error fix for "Expected all tensors to be on the same device" #6

Closed bipinkrish closed 2 weeks ago

bipinkrish commented 9 months ago

Fix for

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_mm)