Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';

解决了题目所述问题。 报错完整信息为: RuntimeError: Expected tensor for argument #1 ‘input’ to have the same device as tensor for argument #2 ‘weight’; but device 1 does not equal 0 (while checking arguments for cudnn_
相关文章
相关标签/搜索