pip install torchsummary
densenet169 = torchvision.models.densenet169(pretrained=True) summary(densenet169, (3, 90, 90))
input_size:channels,height,weight
使用pytorch的，推荐一个工具，pytorchsummary，可以打印模型的结构，模型参数量，和模型的存储大小。
输入特定的输入尺寸[N,C,H,W]，还会显示当前输入尺寸下训练的显存占用，从而选择合适的batch size(N)。
网络搭好了先用这个工具检查一下网络结构，网络容量，显存占用量，可以避免一些低级失误。
