深度学习 lstm

Input and parameter tensors are not at the same device, found input tensor at cpu and parameter tensor at cuda:0

本人在模型里面使用 pytorch官方提供的 lstm接口的时候出现了这个状况。
只需要将lstm放进cuda里面就可以
lstm.to(‘cuda’) ,但是单独测试模型的时候请注释掉这段代码。
或者
if name == ‘main’:
pass
else:
lstm=lstm.to(‘cuda’)