pytorch 一行代碼查看網絡參數總量的實現

更新時間：2021年05月12日 11:21:00 作者：DX_ByYourself

這篇文章主要介紹了pytorch實現一行代碼查看網絡參數總量的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

大家還是直接看代碼吧~

netG = Generator()
print('# generator parameters:', sum(param.numel() for param in netG.parameters()))
netD = Discriminator()
print('# discriminator parameters:', sum(param.numel() for param in netD.parameters()))

補充：PyTorch查看網絡模型的參數量PARAMS和FLOPS等

在PyTorch中，可以使用torchstat這個庫來查看網絡模型的一些信息，包括總的參數量params、MAdd、顯卡內存占用量和FLOPs等。

示例代碼如下：

from torchstat import stat
from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d
model = resnet50()
stat(model, (3, 224, 224))

打印信息如下：

以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章:

python使用pycharm環(huán)境調用opencv庫
這篇文章主要介紹了python使用pycharm環(huán)境調用opencv庫，小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-02-02
python輸出電腦上所有的串口名的方法
今天小編就為大家分享一篇python輸出電腦上所有的串口名的方法，具有好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-07-07
如何利用Python保留指定位數的小數
這篇文章主要給大家介紹了關于如何利用Python保留指定位數的小數,,代碼簡單易懂,非常不錯,對大家學習或者使用Python具有一定的參考借鑒價值,需要的朋友可以參考下
2023-08-08
python面向對象版學生信息管理系統(tǒng)
這篇文章主要為大家詳細介紹了python面向對象版學生信息管理系統(tǒng)，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-06-06
通過Python中的http.server搭建文件上傳下載服務功能
通過本文我們學習了如何使用Python的http.server模塊搭建一個基本的HTTP服務器,并實現文件下載服務,介紹了如何設置服務器端口、自定義文件目錄、定制HTTP響應頭以及處理GET請求,感興趣的朋友跟隨小編一起看看吧
2024-08-08
如何從csv文件構建Tensorflow的數據集
這篇文章主要介紹了如何從csv文件構建Tensorflow的數據集，幫助大家更好的理解和使用Tensorflow，感興趣的朋友可以了解下
2020-09-09
python實現簡單的井字棋游戲(gui界面)
這篇文章主要介紹了python如何實現簡單的井字棋游戲，幫助大家更好的理解和使用python，感興趣的朋友可以了解下
2021-01-01
Atom Python 配置Python3 解釋器的方法
今天小編就為大家分享一篇Atom Python 配置Python3 解釋器的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-08-08
Django+RestFramework API接口及接口文檔并返回json數據操作
這篇文章主要介紹了Django+RestFramework API接口及接口文檔并返回json數據操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
淺談python新手中常見的疑惑及解答
下面小編就為大家?guī)硪黄獪\談python新手中常見的疑惑及解答。小編覺得挺不錯的，現在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2016-06-06