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

[rib12316]

Hi, I'm a beginner and encountered this error when running the code. Does anyone know how to resolve it?

[zhang-xun]

add to before line46 with code:
c2w = c2w.to("cuda:0")

solve the problems.

see solved issue #41