标签：模型 优劣 打印 大小 install models span res style
import models from torchsummary import summary model = models.resnet50(pretrained=False, num_classes=7, scale=1) print(model) summary(model, (3, 640, 640))
使用torchsummay模块，pip install torchsummary
summayr使用方法，加入模型，以及模型输入tensor的形状即可，结尾会给出模型的权重大小，反向传播所需内存大小
使用Print同样可以直接打印模型结构，两者各有优劣。
使用torchsummary打印torch模型结构，包括每层名字以及形状
标签：模型 优劣 打印 大小 install models span res style
原文地址：https://www.cnblogs.com/ywheunji/p/12368667.html
