错误现象
当使用pytorch框架运行代码时,出现RuntimeError: expected type torch.FloatTensor but got torch.cuda.FloatTensor这样的错误
原因分析
是因为计算中有的参数为cuda型,而有的参数却是cpu型,这时就会遇到上述的错误。
使用建议
针对上面的错误,建议在构建模型代码,运行时,将所有的tensor计算进行统一,要么全部指定到device上,要么全部都用cpu,避免混合使用。
当使用pytorch框架运行代码时,出现RuntimeError: expected type torch.FloatTensor but got torch.cuda.FloatTensor这样的错误
是因为计算中有的参数为cuda型,而有的参数却是cpu型,这时就会遇到上述的错误。
针对上面的错误,建议在构建模型代码,运行时,将所有的tensor计算进行统一,要么全部指定到device上,要么全部都用cpu,避免混合使用。