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

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

torchsummary.summary( )中出现了上述错误，torchsummary是应用在pytorch中的一种结构表达方式。

有两种方式可以更改上述错误：

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')

这两种方式都可以避免上述错误

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

参考：
https://blog.csdn.net/u012193416/article/details/87876643