When trying to run the demo I get:
File "/media/ssddata/harald/code/Movement/motion_map/test/pytorch-pose-hg-3d/src/models/hg_3d.py", line 57, in forward
return up1 + up2
RuntimeError: The size of tensor a (5) must match the size of tensor b (4) at non-singleton dimension 2
When trying to run the demo I get: File "/media/ssddata/harald/code/Movement/motion_map/test/pytorch-pose-hg-3d/src/models/hg_3d.py", line 57, in forward return up1 + up2 RuntimeError: The size of tensor a (5) must match the size of tensor b (4) at non-singleton dimension 2
Any idea what I might be doing wrong?