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

floatlazer / semantic_slam

Real time semantic slam in ROS with a hand held RGB-D camera

GNU General Public License v3.0

647 stars 179 forks source link

Closed victor123499 closed 5 years ago

victor123499 commented 6 years ago

I want to train ade20k on pspnet i use your train_cnn, but it appears like this one screenshot from 2018-11-09 14-42-14

can anybody help me?

floatlazer commented 6 years ago

Which version of PyTorch are you using ? This project uses PyTorch 0.4.0.

victor123499 commented 6 years ago

I use PyTorch 0.4.1. Is it ok?

victor123499 commented 6 years ago

what version of cuda and cudnn you use ?

victor123499 commented 6 years ago

i change the pytorch vesion to 0.4.0 but it still have the below problem

screenshot from 2018-11-12 10-45-50