search

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
645 stars 258 forks source link

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

Closed Amore-HDU closed 6 years ago

Amore-HDU commented 6 years ago

Hello, When I run python train.py ./part_A_train_with_val.json ./part_A_val.json 0 ， it occurs RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #3 'other'.

Amore-HDU commented 6 years ago

@leeyeehoo