yzhangcs/post

如何使用GPU加速

monstergazan opened this issue · 0 comments

monstergazan commented

将模型和dataloader的数据放在GPU上，但是一直提示别的张量没有放在GPU，修改了好久都查不出问题。发现问题就在update方法中out = self(x, char_x, lens)部分，一直没能解决