e-lab / LinkNet

168 stars 42 forks source link

Error while running linknet main file #14

Closed Tharun98 closed 6 years ago

Tharun98 commented 6 years ago

Hii, I am getting this error while running main.py RuntimeError: Expected object of type torch.cuda.LongTensor but found type torch.cuda.FloatTensor for argument 2 'target'. Please help me out. Also when i try to run the trained models i am running into error. I am using pytorch to run .net files. I am not able to load them as it is showing error: name cs is not defined. It is a model. Why does it have a variable named cs(here cs represents cityscapes) in it?