在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進(jìn)行特別的計算：
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方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python使用xlrd模塊讀寫Excel文件的方法,較為詳細(xì)的分析了xlrd模塊的安裝、使用與操作Excel文件的相關(guān)技巧,需要的朋友可以參考下2015-05-05
Python中使用 Selenium 實現(xiàn)網(wǎng)頁截圖實例
這篇文章主要介紹了Python中使用 Selenium 實現(xiàn)網(wǎng)頁截圖實例,Selenium支持Java、C#、Ruby 以及 Python等語言,本文以Python語言為例,需要的朋友可以參考下2014-07-07
如何將matlab數(shù)據(jù)導(dǎo)入到Python中使用
這篇文章主要介紹了如何將matlab數(shù)據(jù)導(dǎo)入到Python中使用,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-12-12
Python 字符串操作(string替換、刪除、截取、復(fù)制、連接、比較、查找、包含、大小寫轉(zhuǎn)換、分割等)
這篇文章主要介紹了Python 字符串操作(string替換、刪除、截取、復(fù)制、連接、比較、查找、包含、大小寫轉(zhuǎn)換、分割等),需要的朋友可以參考下2018-03-03
Python re.findall中正則表達(dá)式(.*?)和參數(shù)re.S使用
本文主要介紹了Python re.findall中正則表達(dá)式(.*?)和參數(shù)re.S使用，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
這篇文章主要介紹了Python的列表和列表算法，小編感覺這篇文章具有一定參考價值，需要的朋友可以了解下，希望能給你帶來幫助2021-08-08
這篇文章主要介紹了Python中hashlib模塊的摘要算法詳解,摘要算法又稱哈希算法、散列算法,它通過一個函數(shù)，把任意長度的數(shù)據(jù)轉(zhuǎn)換為一個長度固定的數(shù)據(jù)串,通常用16進(jìn)制的字符串表示,需要的朋友可以參考下2023-08-08
python之json文件轉(zhuǎn)xml文件案例講解
這篇文章主要介紹了python之json文件轉(zhuǎn)xml文件案例講解,本篇文章通過簡要的案例,講解了該項技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-08-08
這篇文章主要給大家介紹了關(guān)于Python利用pip安裝tar.gz格式的離線資源包的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
當(dāng)大家的站點越來越來越多的時候會發(fā)現(xiàn)管理起來也挺復(fù)雜的，所以這篇文章給大家分享下利用python批量檢查網(wǎng)站的可用性的功能，對大家管理網(wǎng)站具有很實用的價值，有需要的朋友可以參考借鑒。2016-09-09