常见错误 RuntimeError: expected type torch.FloatTensor but got torch.cuda.FloatTensor
摘要:
https://www.jianshu.com/p/0be7a375bdbe https://blog.csdn.net/qq_38410428/article/details/82973895 计算中有的参数为cuda型,有的参数却是cpu型,就会遇到这样的错误。 解决办法: 该加.cuda()的 阅读全文
posted @ 2019-09-02 22:20 吱吱了了 阅读(4699) 评论(0) 推荐(0)
浙公网安备 33010602011771号