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