expected FloatTensor but found cuda.FloatTensor

[viig99]

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

Got this error on line 407 of efficientdensenet_crnn/train/pytorch-train/models/efficient_densecrnn.py, can you please help ?

[phybrain]

can you provide more details? did you use model.cuda()

[windowxiaoming]

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

Got this error on line 407 of efficientdensenet_crnn/train/pytorch-train/models/efficient_densecrnn.py, can you please help ?

This code is based on pytorch=0.2.0，so the bug is appearance。So the author can modify the code so that it can run which the pytorch=0.4.0 or pytorch=0.4.1 ，Thanks

[wjw-2017]

''' RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'source' ''' Got this error on line 407 of efficientdensenet_crnn/train/pytorch-train/models/efficient_densecrnn.py, can you please help ?

This code is based on pytorch=0.2.0，so the bug is appearance。So the author can modify the code so that it can run which the pytorch=0.4.0 or pytorch=0.4.1 ，Thanks

I also met this problem, have you solved it？？？ can you help me???