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