search

phybrain / efficientdensenet_crnn

memory efficient densenet+lstm+ctc实现中文识别
MIT License
31 stars 9 forks source link

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

Open wjw-2017 opened 4 years ago

wjw-2017 commented 4 years ago

''' 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

_Originally posted by @windowxiaoming in https://github.com/phybrain/efficientdensenet_crnn/issues/2#issuecomment-437740805_