fvcore计算模型参数量和计算量
fvcore是一个易用的工具。fvcore是Facebook开源的一个轻量级的核心库,它提供了各种计算机视觉框架中常见且基本的功能。其中就包括了统计模型的参数以及FLOPs等。
pip install fvcore
- FLOPS:注意全大写,是floating point operations per second的缩写,意指每秒浮点运算次数,理解为计算速度。是一个衡量硬件性能的指标。
- FLOPs:注意s小写,是floating point operations的缩写(s表复数),意指浮点运算数,理解为计算量。可以用来衡量算法/模型的复杂度。
1.计算量
import torch from torchvision.models import resnet50 from fvcore.nn import FlopCountAnalysis, parameter_count_table # 创建resnet50网络 model = resnet50(num_classes=1000) # 创建输入网络的tensor tensor = (torch.rand(1, 3, 224, 224),) # 分析FLOPs flops = FlopCountAnalysis(model, tensor) print("FLOPs: ", flops.total())
2.参数量
import torch from torchvision.models import resnet50 from fvcore.nn import FlopCountAnalysis, parameter_count_table # 创建resnet50网络 model = resnet50(num_classes=1000) # 分析parameters print(parameter_count_table(model))
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 没有源码,如何修改代码逻辑?
· PowerShell开发游戏 · 打蜜蜂
· 在鹅厂做java开发是什么体验
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战
2021-04-15 第二题 既约分数
2021-04-15 2020蓝桥杯第二场省赛 第一题
2020-04-15 MFC直方图