import torch              
from torchstat import stat           
import torchvision.models as models      
net = models.vgg11()           
stat(net,(3,224,224))    # (3,224,224)表示输入图片的尺寸

使用 torchstat库

结果为：