pytorch问题汇总
仅用于记录平时碰到的问题。
torch与torchvision版本不对应
极智AI | torch与torchvision版本对应关系速查 - 知乎 (zhihu.com)
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor
不要直接tensor.to，要tensor=tensor.to，见问题解决：Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the-CSDN博客
本博客所有文章除特别声明外，均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 雨白的博客小屋！
评论
