torch.FloatTensor Error

felixchenfy / Detect-Hand-Grasping-Object

A toy project: Detect my hand grasping object in the video. Backbone algorithms: SiamMask, Mask_RCNN, OpenPose

13 stars 2 forks source link

torch.FloatTensor Error #1

Open HHIR opened 5 years ago

HHIR commented 5 years ago

Dear @felixchenfy ,

that is interesting project, I tired to run "python3 main_detect_hand_grasping_object.py" but Im getting this error:

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

Do you have any Idea to fix it? Thanks in advance.

felixchenfy commented 5 years ago

Hi, it's probably due to a mismatch of pytorch's version. I didn't encounter this problem before, so I don't know how to solve it. Sorry.
When I was doing this project, I just followed the installation instructions of the README of the three projects that I referenced.
Good luck. Thanks