from torchsummary import summary
summary(model,(3, 224, 224),batch_size=1)
报错
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type
(torch.FloatTensor) should be the same
解决方法:在model后面加".cuda()"
summary(model.cuda(),(3, 224, 224),batch_size=1)