在Pytorch中计算自己模型的FLOPs方式
更新时间：2019年12月30日 09:53:17 作者：2Dposeliu
今天小编就为大家分享一篇在Pytorch中计算自己模型的FLOPs方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
https://github.com/Lyken17/pytorch-OpCounter
安装方法很简单：
pip install thop
基本用法：
from torchvision.models import resnet50from thop import profile model = resnet50() flops, params = profile(model, input_size=(1, 3, 224,224))
对自己的module进行特别的计算：
class YourModule(nn.Module):
# your definition
def count_your_model(model, x, y):
# your rule
hereflops, params = profile(model, input_size=(1, 3, 224,224),
custom_ops={YourModule: count_your_model})
以上这篇在Pytorch中计算自己模型的FLOPs方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python使用xlrd模块读写Excel文件的方法,较为详细的分析了xlrd模块的安装、使用与操作Excel文件的相关技巧,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了Python中使用 Selenium 实现网页截图实例,Selenium支持Java、C#、Ruby 以及 Python等语言,本文以Python语言为例,需要的朋友可以参考下2014-07-07
这篇文章主要介绍了如何将matlab数据导入到Python中使用,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-12-12
Python 字符串操作(string替换、删除、截取、复制、连接、比较、查找、包含、大小写转换、分割等)
这篇文章主要介绍了Python 字符串操作(string替换、删除、截取、复制、连接、比较、查找、包含、大小写转换、分割等),需要的朋友可以参考下2018-03-03
Python re.findall中正则表达式(.*?)和参数re.S使用
本文主要介绍了Python re.findall中正则表达式(.*?)和参数re.S使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
这篇文章主要介绍了Python的列表和列表算法，小编感觉这篇文章具有一定参考价值，需要的朋友可以了解下，希望能给你带来帮助2021-08-08
这篇文章主要介绍了Python中hashlib模块的摘要算法详解,摘要算法又称哈希算法、散列算法,它通过一个函数，把任意长度的数据转换为一个长度固定的数据串,通常用16进制的字符串表示,需要的朋友可以参考下2023-08-08
这篇文章主要介绍了python之json文件转xml文件案例讲解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下2021-08-08
这篇文章主要给大家介绍了关于Python利用pip安装tar.gz格式的离线资源包的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-09-09
当大家的站点越来越来越多的时候会发现管理起来也挺复杂的，所以这篇文章给大家分享下利用python批量检查网站的可用性的功能，对大家管理网站具有很实用的价值，有需要的朋友可以参考借鉴。2016-09-09
最新评论