pytorch 一行代碼查看網(wǎng)絡(luò)參數(shù)總量的實現(xiàn)
更新時間：2021年05月12日 11:21:00 作者：DX_ByYourself
這篇文章主要介紹了pytorch實現(xiàn)一行代碼查看網(wǎng)絡(luò)參數(shù)總量的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
大家還是直接看代碼吧~
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()))
補(bǔ)充：PyTorch查看網(wǎng)絡(luò)模型的參數(shù)量PARAMS和FLOPS等
在PyTorch中，可以使用torchstat這個庫來查看網(wǎng)絡(luò)模型的一些信息，包括總的參數(shù)量params、MAdd、顯卡內(nèi)存占用量和FLOPs等。
示例代碼如下：
from torchstat import stat from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d model = resnet50() stat(model, (3, 224, 224))
打印信息如下：
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
python使用pycharm環(huán)境調(diào)用opencv庫
這篇文章主要介紹了python使用pycharm環(huán)境調(diào)用opencv庫，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-02-02
今天小編就為大家分享一篇python輸出電腦上所有的串口名的方法，具有好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
這篇文章主要給大家介紹了關(guān)于如何利用Python保留指定位數(shù)的小數(shù),,代碼簡單易懂,非常不錯,對大家學(xué)習(xí)或者使用Python具有一定的參考借鑒價值,需要的朋友可以參考下2023-08-08
python面向?qū)ο蟀鎸W(xué)生信息管理系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了python面向?qū)ο蟀鎸W(xué)生信息管理系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-06-06
通過Python中的http.server搭建文件上傳下載服務(wù)功能
通過本文我們學(xué)習(xí)了如何使用Python的http.server模塊搭建一個基本的HTTP服務(wù)器,并實現(xiàn)文件下載服務(wù),介紹了如何設(shè)置服務(wù)器端口、自定義文件目錄、定制HTTP響應(yīng)頭以及處理GET請求,感興趣的朋友跟隨小編一起看看吧2024-08-08
如何從csv文件構(gòu)建Tensorflow的數(shù)據(jù)集
這篇文章主要介紹了如何從csv文件構(gòu)建Tensorflow的數(shù)據(jù)集，幫助大家更好的理解和使用Tensorflow，感興趣的朋友可以了解下2020-09-09
這篇文章主要介紹了python如何實現(xiàn)簡單的井字棋游戲，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2021-01-01
今天小編就為大家分享一篇Atom Python 配置Python3 解釋器的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
Django+RestFramework API接口及接口文檔并返回json數(shù)據(jù)操作
這篇文章主要介紹了Django+RestFramework API接口及接口文檔并返回json數(shù)據(jù)操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-07-07
下面小編就為大家?guī)硪黄獪\談python新手中常見的疑惑及解答。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-06-06