使用torchsummary打印torch模型结构,包括每层名字以及形状
Posted ywheunji
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用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同样可以直接打印模型结构,两者各有优劣。
以上是关于使用torchsummary打印torch模型结构,包括每层名字以及形状的主要内容,如果未能解决你的问题,请参考以下文章
PyTorch打印模型结构输出维度和参数信息(torchsummary)
错误处理:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be