JimmySuen / integral-human-pose

Integral Human Pose Regression
MIT License
471 stars 76 forks source link

RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.cuda.LongTensor for argument #2 'other' #4

Closed sjtuytc closed 5 years ago

sjtuytc commented 5 years ago

How do I fix this error? It occurs at common_pytorch/common_loss/integral.py Line17

lck1201 commented 5 years ago

@sjtuytc Hi, the code runs properly on our machine. Please check you have installed everything correctly.

xiangdonglai commented 5 years ago

@lck1201 Hello. I also encounter this problem. May I know the PyTorch version that you use?

lck1201 commented 5 years ago

@xiangdonglai I install via conda install pytorch torchvision -c pytorch, version 0.4.0

xiangdonglai commented 5 years ago

@lck1201 Thank you for your response. I also solve the problem using Python 3.6 and PyTorch 0.4.0.

lck1201 commented 5 years ago

@sjtuytc please follow @xiangdonglai to install requirements. Now close issue.. If you have further question, I would re-open it.