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