 

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

2019-12-22 分类：Pytorch学习评论(0) 赞(389)

使用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'

赞(389) 打赏

版权声明：本文采用知识共享署名4.0国际许可协议 [BY-NC-SA] 进行授权
文章名称：《[mcj]RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same》
文章链接：https://www.machunjie.com/deeplearning/pytorch/627.html
本站资源仅供个人学习交流，请于下载后24小时内删除，不允许用于商业用途，否则法律问题自行承担。

分享到

pytorch