首页 > TAG信息列表 > TorchSummary
TorchSummary无法载入Dict数据类型解决方法
前言 torchsummary是一个比较不错的评价网络数据结构的包，但是目前torchsummary只支持网络输入和输出为torch.Tensor类型的网络，在对一些较为复杂的网络模型中，载入的可能并不一定为tensor类型，也有可能是list或者dict类型的数据。 为了可以支持自己的网络结构，为此简单修改了一下torch使用torchsummary时报错AttributeError: ‘list‘ object has no attribute ‘size‘
使用torchsummary时报错AttributeError: 'list' object has no attribute 'size' 说明使用代码报错截图查明原因解决方法成功输出模型最后 说明 因为最近刚开始学pytorch，想输出模型结果来看看，但是他并没有像keras那么简单，就挺苦恼的。但学习的过程从来都不会一帆风顺的，加使用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的形状即可，结尾pytorch实用工具：torchsummary
一. torchsummmary工具： 其主要是用来计算网络的计算参数等信息的，其大概的流程为： 安装指令： pip install torchsummary 使用方法： import torch, torchvision model = torchvision.models.vgg model = torchvision.models.vgg16() from torchsummary import summary