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

Pytorch使用过程中

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

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