torchsummary

首页 > 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