[mcj]RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

发表日期：2019-12-22 马春杰杰分类：Pytorch学习评论(0) 本文内容有更新，最后修改日期为：2019年12月25日

使用pytorch的时候，想要对网络进行可视化，结果报错为：

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

这个是使用GPU 的问题，两种修改方法：

if __name__ == '__main__':
    model = fishnet99()
    torchsummary.summary(model.cuda(), (3, 224, 224))

if __name__ == '__main__':

model = fishnet99()

torchsummary.summary(model.cuda(), (3, 224, 224))

if __name__ == '__main__':
    model = fishnet99()
    torchsummary.summary(model, (3, 224, 224),device='cpu')

if __name__ == '__main__':

model = fishnet99()

torchsummary.summary(model, (3, 224, 224),device='cpu')

第一种就是 convert your network to cuda，第二种就是 call torchsummary.summary with device='cpu'

如果你对这篇文章有什么疑问或建议，欢迎下面留言提出，我看到会立刻回复！

打赏

标签：pytorch

好好学习，天天向上。加油！！！

马春杰杰博客始建于2017年6月14日，博客主要分享网站建设中遇到的问题及解决方案、自己在读书过程中的心得体会及分享一些自觉有意义的音视频内容，记录些生活中的琐事，希望博客能督促怠惰的自己不断学习，不断进步。

不公开

※ 微博 ※

不公开

※ 支付宝好友 ※

不公开

※ 个人QQ号 ※

总金额：1.00 马杜币