栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Flops 计算以及代码快速查看网络参数、浮点数

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Flops 计算以及代码快速查看网络参数、浮点数

详细计算:卷积神经网络CNN中的参数量(parameters)和计算量(FLOPs ) - 知乎

FLOPS: 全大写,指每秒浮点运算次数,可以理解为计算的速度。是衡量硬件性能的一个指标。(硬件) 

FLOPs: s小写,指浮点运算数,理解为计算量。可以用来衡量算法/模型的复杂度。(模型) 在论文中常用GFLOPs(1 GFLOPs = 10^9 FLOPs)

安装方式: 

pip install torchstat

 使用方法:

from torchstat import stat
import torchvision.models as models

model = models.alexnet()
stat(model, (3, 224, 224))

用来计算pytorch构建的网络的参数,空间大小,MAdd,FLOPs等指标; 

使用torchsummary 计算参数:

pip install torchsummary

import torch
import torchvision.models as models
model = models.alexnet()
from torchsummary import summary
summary(model, (3, 224, 224))

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/343957.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号