pytorch获取模型的参数量和运算量

最新推荐文章于 2024-07-30 15:16:38 发布
fancyNSEU 最新推荐文章于 2024-07-30 15:16:38 发布
阅读量1.3k 收藏 11
点赞数 1
文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接： https://blog.csdn.net/yule200910/article/details/127812853
版权

定义模型

net = nn.Sequential(
    nn.Linear(28 * 28, 400),
    nn.ReLU(),
    nn.Linear(400, 200),
    nn.ReLU(),
    nn.Linear(200, 100),
    nn.ReLU(),
    nn.Linear(100, 10)
).cuda()

1. torch自身方法获取参数量

total = sum([param.nelement() for param in net.parameters()])
print("Number of parameter: %.2fM" % (total / 1e6))

2. torchsummary库获取模型的参数量

from torchsummary import summary

summary(net, input_size=(784,))

3. thop库获取模型参数量和计算量

from thop import profile, clever_format


myinput = torch.zeros((1, 1, 784)).cuda()
flops, params = profile(net, inputs=myinput)
flops, params = clever_format([flops, params], "%.3f")
print(flops, params)

分析：

这个模型参数量为415.31K，但是浮点计算量为414.6K，因为ReLU等层不需要浮点计算

fancyNSEU
关注 关注
  • 1
    点赞
  • 踩
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Pytorch的卷积神经网络参数详解与模型demo
09-15
### PyTorch卷积神经网络参数详解与模型示例 #### 一、PyTorch一维卷积神经网络参数解析 PyTorch是一门强大的深度学习框架，它提供了丰富的API来构建复杂的神经网络模型。本篇文章主要聚焦于一维卷积神经网络(Conv...
pytorch统计模型参数量并输出
xidaoliang的博客
05-18 2518
pytorch模型参数统计
pytorch 计算网络模型的计算量FLOPs和参数量parameter之殊途同归
weixin_40893448的博客
04-27 4653
在进行论文撰写时，我们通常要通过计算网络模型的计算量FLOPs和参数量parameter来评估模型的性能，本文总结了几种常用的计算方式，大家可以尝试一下。为了能够便于读者理解，我们选取pytorch自带的网络resnet34进行测试，也可自行更改为其他或所提网络。
Pytorch代码：打印模型每层的参数数量和总参数量
Never Give Up
07-30 458
【代码】Pytorch代码：打印模型每层的参数数量和总参数量。
python-基础-计算模型参数总量
SKY_yiyi_9的博客
08-16 8037
计算模型参数总量 方法1：pytorch自带方法，计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print(“Number of parameter: %.2fM” % (total/1e6)) 例子： 红色断点放到下一行，并用debug运行。结果为： 方法2：用工具包thop 要安装：pip install thop 在程序里输入： from thop import profile from thop
5种方法获取Torch网络模型参数量计算量等信息
热门推荐
十二月未央未眠
04-19 3万+
目录模型参数量和计算量是什么为什么要统计模型参数量和计算量常见的模型参数量和计算量的计算方法代码方法1，利用torch自带方法，自写函数方法2，torchsummary.summary方法3 torchstat.stat方法4 thop.profile方法5 ptflops.get_model_complexity_info 模型参数量和计算量是什么 计算量是指网络模型需要计算的运算次数，参数量是指网络模型自带的参数数量多少 计算量对应时间复杂度，参数量对应于空间复杂度 计算量决定了网络执行时间的长短，参
pytorch可视化输出模型计算力和参数量，设备损耗，运行时间等信息
微信号：Custom-Software
12-10 1769
输出下面的表格形式的，将自己的模型都输出出来。 from torchstat import stat from torchvision.models import resnet50 model = resnet50() stat(model, (3, 224, 224)) # (3,224,224)表示输入图片的尺寸 记得： pip install torchstat 不过好像会报下面的错误： 所以，只需要在最下面的错，点进去了。将detach()后面变为detach().cpu,bu，不过还是会报
Python-PyTorch神经网络模型分析器参数规模FLOPsMAdd内存消耗
08-09
本主题将深入探讨如何使用PyTorch的工具来分析神经网络模型的参数规模、运算量（FLOPs和MAdd）以及内存消耗。这些指标对于理解模型的复杂性、优化性能以及资源管理至关重要。 首先，我们要了解模型参数规模。参数...
pytorch介绍.zip
03-18
在模型保存与加载方面，PyTorch提供了`torch.save()`和`torch.load()`函数，可以将模型的参数和状态持久化到磁盘，以便于后续使用。`torch.jit`模块则支持模型的脚本化和编译，以提高推理速度。 总的来说，PyTorch...
EffectivePyTorch：PyTorch教程和最佳实践
02-03
5. **优化器（Optimizer）**：PyTorch提供了多种优化算法，如SGD（随机梯度下降）、Adam等，用于更新模型参数以最小化损失函数。 6. **数据加载器（DataLoader）**：`torch.utils.data.DataLoader`负责从数据集加载...
手写RNN Pytorch 实现 包含数据集和代码 可直接运行.zip
08-27
自动梯度系统则用于计算模型参数的梯度，以便于反向传播和优化过程。 2. **循环神经网络（RNN）**：RNNs是一种能够处理序列输入的神经网络结构。每个时间步，RNN接收输入并更新其内部状态，该状态包含了前面所有...
pytorch得到模型的计算量和参数量
qq_35407318的博客
10-29 1万+
文章目录方法1方法2方法3方法4 方法1 pytorch自带方法，计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print("Number of parameter: %.2fM" % (total/1e6)) 方法2 计算模型参数总量和模型计算量 def count_params(model, input_size=224): # param_sum = 0 with open('
PyTorch中查看模型的参数信息的几种方式
长夜漫漫，无心睡眠
01-25 3340
【代码】PyTorch中查看模型的参数信息的几种方式。
Pytorch：打印模型中每一层的输出大小shape与参数量
weixin_42233605的博客
06-10 7411
在使用Pytorch编写模型时，为了验证检查代码是否正确，需要查看模型每一层的输出shape或是模型结构。 然而在Pytorch只能够直接查看模型的结构，无法简单、直观地查看每一层的输出大小，需要自行手动计算，并且在输入shape改变时，每一层的输出同样随之变化，需要重复计算。 因此，这里建议使用一个工具，torchsummary...
pytorch打印模型结构和参数
weixin_39107270的博客
10-30 3751
当我们使用pytorch进行模型训练或测试时，有时候希望能知道模型每一层分别是什么，具有怎样的参数。此时我们可以将模型打印出来，输出每一层的名字、类型、参数等。
计算pytorch模型算力和参数的大小
ze1336365763的博客
10-17 1935
设计神经网络时需要根据硬件考虑模型参数量和算力，故需要计算神经网络模型的params和Flops，python的三方库ptflops就可以很好的做这件事，show code： import torch.nn as nn import torch from ptflops import get_model_complexity_info class Net(nn.Module): def __init__(self,in_c,class_num): super(Net, self)
深度学习卷积网络浮点计算量和参数量的计算（附Pytorch代码）
limingmin2020的博客
05-24 1353
本文作者介绍了深度学习卷积网络的浮点计算量和参数量的计算，还附上了以Pytorch代码框架示例的案例。非常适合大家学习了解，值得一读！
6种方法计算神经网络参数量Params、计算量FLOPs、Macs简单代码
qq_40193227的博客
11-08 4598
6种方法计算神经网络参数量Params、计算量FLOPs、Macs简单代码
PyTorch 打印模型结构、输出维度和参数信息（torchsummary）
梁小憨憨的博客
01-27 1万+
使用 PyTorch 深度学习搭建模型后，如果想查看模型结构，可以直接使用 print(model) 函数打印。但该输出结果不是特别直观，查阅发现有个能输出类似 keras 风格 model.summary() 的模型可视化工具。这里记录一下方便以后查阅。
pytorch获取模型的运算量
最新发布
08-27
在PyTorch中，获取模型的运算量，通常是通过计算浮点运算次数（FLOPs，Floating Point Operations）来实现的。FLOPs是衡量模型复杂度的一个重要指标，它表示模型在进行一次完整的前向传播过程中所需的浮点运算总数。PyTorch提供了多种方式来估算或计算模型的运算量，包括： 1. **使用`thop`库**: `thop`是一个专门用于计算PyTorch模型FLOPs和参数量的库。通过导入`thop`库和调用其相应的函数，可以方便地获得模型的运算量。 2. **使用`ptflops`库**: 这是另一个用于计算PyTorch模型复杂度的库。通过`ptflops.get_model_complexity_info`方法，可以直接得到模型的FLOPs和参数量。 3. **使用`torchstat`库**: 该库提供了一个快速估计FLOPs的方法。首先需要将模型实例化为一个`torch.nn.Module`对象，然后使用`FLOPsCounter`类的`count_flops_param`方法来获取模型的FLOPs。 4. **使用`torchsummary`包**: 通过`torchsummary`包中的`summary()`函数，可以得到模型的详细计算量报告，包括每层的输入输出维度、FLOPs以及参数量。 5. **自定义计算**: 对于一些简单的模型或特定的层，也可以通过自定义代码来计算FLOPs。这通常需要遍历模型的每一层，根据层的参数和操作类型手动计算FLOPs。 不同的方法有着不同的易用性和灵活性。`thop`和`ptflops`库提供了较为简洁的接口，适合快速获取常见模型的运算量。而`torchstat`和`torchsummary`则在提供运算量信息的同时，还能给出更多模型性能相关的数据。自定义计算方法虽然最为灵活，但实现起来也最为复杂。
写文章

热门文章

  • opencv-C++调用torch-python训练好的网络 4534
  • 图像分类、目标检测、图像分割----简介 2935
  • Qt5.12用cout输出 1361
  • pytorch获取模型的参数量和运算量 1310
  • ubuntu修改默认python版本 1262

分类专栏

  • 常用资源连接 2篇

最新文章

  • A Survey of multi source information fusion
  • Linux下pip安装的包不在当前conda环境目录下
  • pytroch模型可视化
2024年1篇
2023年1篇
2022年6篇
2021年20篇

目录

目录

评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

fancyNSEU

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或 充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值

玻璃钢生产厂家大连玻璃钢雕塑价格陕西玻璃钢人物动物雕塑厂家南京玻璃钢景观雕塑供货商舟山创意玻璃钢雕塑销售电话漳州玻璃钢人物雕塑定制哪有卖玻璃钢雕塑无锡灰色玻璃钢花盆新郑玻璃钢浮雕抽象景观雕塑厂家玻璃钢彩色雕塑深圳玻璃钢大象雕塑安徽现代人物玻璃钢雕塑开封不锈钢玻璃钢雕塑定做厂家江西玻璃钢雕塑设计公司玻璃钢寺庙雕塑施工图玻璃钢动物雕塑园林小品浙江超市商场美陈福州水果玻璃钢雕塑供应商山西玻璃钢酒店人物雕塑发光玻璃钢雕塑加工厂玻璃钢雕塑西游记人物玻璃钢广场雕塑设计价格设计商场美陈多少钱杭州个性化玻璃钢雕塑安宁市玻璃钢雕塑设计定制菏泽玻璃钢雕塑制作厂家海南白沙玻璃钢海洋生物动物雕塑北京市商场美陈哪家公司好南京人物玻璃钢雕塑公司深圳佛像玻璃钢雕塑定制邯郸景观玻璃钢雕塑香港通过《维护国家安全条例》两大学生合买彩票中奖一人不认账让美丽中国“从细节出发”19岁小伙救下5人后溺亡 多方发声单亲妈妈陷入热恋 14岁儿子报警汪小菲曝离婚始末遭遇山火的松茸之乡雅江山火三名扑火人员牺牲系谣言何赛飞追着代拍打萧美琴窜访捷克 外交部回应卫健委通报少年有偿捐血浆16次猝死手机成瘾是影响睡眠质量重要因素高校汽车撞人致3死16伤 司机系学生315晚会后胖东来又人满为患了小米汽车超级工厂正式揭幕中国拥有亿元资产的家庭达13.3万户周杰伦一审败诉网易男孩8年未见母亲被告知被遗忘许家印被限制高消费饲养员用铁锨驱打大熊猫被辞退男子被猫抓伤后确诊“猫抓病”特朗普无法缴纳4.54亿美元罚金倪萍分享减重40斤方法联合利华开始重组张家界的山上“长”满了韩国人？张立群任西安交通大学校长杨倩无缘巴黎奥运“重生之我在北大当嫡校长”黑马情侣提车了专访95后高颜值猪保姆考生莫言也上北大硕士复试名单了网友洛杉矶偶遇贾玲专家建议不必谈骨泥色变沉迷短剧的人就像掉进了杀猪盘奥巴马现身唐宁街 黑色着装引猜测七年后宇文玥被薅头发捞上岸事业单位女子向同事水杯投不明物质凯特王妃现身！外出购物视频曝光河南驻马店通报西平中学跳楼事件王树国卸任西安交大校长 师生送别恒大被罚41.75亿到底怎么缴男子被流浪猫绊倒 投喂者赔24万房客欠租失踪 房东直发愁西双版纳热带植物园回应蜉蝣大爆发钱人豪晒法院裁定实锤抄袭外国人感慨凌晨的中国很安全胖东来员工每周单休无小长假白宫：哈马斯三号人物被杀测试车高速逃费 小米：已补缴老人退休金被冒领16年 金额超20万

玻璃钢生产厂家 XML地图 TXT地图 虚拟主机 SEO 网站制作 网站优化