> Keras style model.summary() in PyTorch
> torch-summary 1.3.3
以上两个链接里,我需要的是多个输入Multiple Inputs w/ Different Data Types
from torchsummary import summary
summary(model, [(1, 2048), (1, 300)], batch_size=-1, device='cuda')
import torch
import torchvision
from torchsummary import summary #使用 pip install torchsummary
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
vgg = torchvision.models.vgg16().to(device)
summary(vgg, input_size=(3, 224, 224))