模型合并好了，跑示例代码报错了

[xjlv1113]

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)a'

[nuaabuaa07]

我报的错误和你相似，区别是在发现，在两个显卡上，不允许。还没找到解决方法