model.summary() | torch summary 画模型图

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

> 在Pytorch上使用summary

> 使用torchsummary打印torch模型的每层形状参数