其他分享
首页 > 其他分享> > 使用torchsummary打印torch模型结构,包括每层名字以及形状

使用torchsummary打印torch模型结构,包括每层名字以及形状

作者:互联网

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同样可以直接打印模型结构,两者各有优劣。

标签:models,模型,torch,每层,summary,import,model,torchsummary
来源: https://www.cnblogs.com/ywheunji/p/12368667.html