在Pytorch中計(jì)算自己模型的FLOPs方式
更新時(shí)間：2019年12月30日 09:53:17 作者：2Dposeliu
今天小編就為大家分享一篇在Pytorch中計(jì)算自己模型的FLOPs方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
https://github.com/Lyken17/pytorch-OpCounter
安裝方法很簡(jiǎn)單：
pip install thop
基本用法：
from torchvision.models import resnet50from thop import profile model = resnet50() flops, params = profile(model, input_size=(1, 3, 224,224))
對(duì)自己的module進(jìn)行特別的計(jì)算：
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中計(jì)算自己模型的FLOPs方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python 代碼在函數(shù)中運(yùn)行得更快的原因解析
我們知道,python 是一種解釋型語(yǔ)言,它會(huì)逐行讀取并執(zhí)行代碼,小伙伴們可能會(huì)有這個(gè)疑問：為什么在函數(shù)中運(yùn)行的 Python 代碼速度更快,今天這篇文章將會(huì)解答大家心中的疑惑2023-09-09
這篇文章主要介紹了python Timer 類使用介紹的相關(guān)資料，幫助大家更好的理解和使用python iupdatable包，感興趣的朋友可以了解下2020-12-12
這篇文章主要介紹了Python networkx中獲取圖的鄰接矩陣方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-12-12
這篇文章主要介紹了Python測(cè)試模塊doctest使用解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08
介紹Python的Django框架中的靜態(tài)資源管理器django-pipeline
這篇文章主要介紹了介紹Python的Django框架中的靜態(tài)資源管理器django-pipeline,django-pipeline是一個(gè)開源項(xiàng)目,被用來處理css等靜態(tài)文件,需要的朋友可以參考下2015-04-04
Pandas中的unique()和nunique()區(qū)別詳解
Pandas中Series和DataFrame的兩種數(shù)據(jù)類型中都有nunique()和unique()方法，本文詳細(xì)的介紹了兩者的區(qū)別，具有一定的參考價(jià)值，感興趣的可以了解一下2022-08-08
Python+Pyecharts實(shí)現(xiàn)散點(diǎn)圖的繪制
散點(diǎn)圖是指在回歸分析中，數(shù)據(jù)點(diǎn)在直角坐標(biāo)系平面上的分布圖，散點(diǎn)圖表示因變量隨自變量而變化的大致趨勢(shì)，據(jù)此可以選擇合適的函數(shù)對(duì)數(shù)據(jù)點(diǎn)進(jìn)行擬合。本文將利用Python Pyecharts實(shí)現(xiàn)散點(diǎn)圖的繪制，需要的可以參考一下2022-06-06
基于Python+Pygame實(shí)現(xiàn)經(jīng)典賽車游戲
這篇文章主要為大家分享了一個(gè)基于Python和Pygame實(shí)現(xiàn)的賽車小游戲，文中的示例代碼講解詳細(xì)，對(duì)我們學(xué)習(xí)Python有一定幫助，需要的可以參考一下2022-04-04
Pythonic版二分查找實(shí)現(xiàn)過程原理解析
這篇文章主要介紹了Pythonic版二分查找實(shí)現(xiàn)過程原理解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-08-08
python用類實(shí)現(xiàn)文章敏感詞的過濾方法示例
這篇文章主要介紹了python用類實(shí)現(xiàn)文章敏感詞的過濾方法示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10