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中進度條tqdm包使用方法及特性,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
這篇文章主要介紹了 python基礎教程之匿名函數lambda的相關資料,需要的朋友可以參考下2017-01-01
這篇文章主要給大家介紹了關于python中os和sys模塊的區(qū)別與常用方法的相關資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧。2017-11-11
在日常編程中,我們經常會遇到需要將二進制文件轉換為文本文件的情況,在Python中,我們可以利用各種庫和技術來完成這項任務,本文將介紹如何使用Python將二進制文件轉換為文本文件,并提供實用的代碼示例,需要的朋友可以參考下2024-04-04
這篇文章主要介紹了python scrapy腳本報錯問題及解決,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
下面小編就為大家?guī)硪黄鉀QPython 遍歷字典時刪除元素報異常的問題。小編覺得挺不錯的，現在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-09-09
這篇文章主要介紹了Pandas庫之DataFrame使用的學習筆記，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-06-06
在我們使用Python語言進行機器學習編程的時候，這是一個非常常用的基礎庫。本文針對Python 機器學習庫 NumPy入門教程，感興趣的朋友跟隨腳本之家小編一起學習吧2018-04-04
這篇文章主要介紹了python 列表推導和生成器表達式的使用方法，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2021-02-02
這篇文章主要介紹了python學習之讀取配置文件，文章基于python的相關資料展開對主題的詳細介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-04-04