在pytorch中查看可訓(xùn)練參數(shù)的例子
更新時間：2019年08月18日 10:56:50 作者：玩人
今天小編就為大家分享一篇在pytorch中查看可訓(xùn)練參數(shù)的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch中我們有時候可能需要設(shè)定某些變量是參與訓(xùn)練的，這時候就需要查看哪些是可訓(xùn)練參數(shù)，以確定這些設(shè)置是成功的。
pytorch中model.parameters()函數(shù)定義如下：
def parameters(self):
r"""Returns an iterator over module parameters.
This is typically passed to an optimizer.
Yields:
Parameter: module parameter
Example::
>>> for param in model.parameters():
>>> print(type(param.data), param.size())
<class 'torch.FloatTensor'> (20L,)
<class 'torch.FloatTensor'> (20L, 1L, 5L, 5L)
"""
for name, param in self.named_parameters():
yield param
所以，我們可以遍歷named_parameters()中的所有的參數(shù)，只打印那些param.requires_grad=True的變量。具體實現(xiàn)代碼如下所示：
for name, param in model.named_parameters():
if param.requires_grad:
print(name)
這樣打印出的結(jié)果就是模型中所有的可訓(xùn)練參數(shù)列表！
以上這篇在pytorch中查看可訓(xùn)練參數(shù)的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python每5分鐘從kafka中提取數(shù)據(jù)的例子
今天小編就為大家分享一篇python每5分鐘從kafka中提取數(shù)據(jù)的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
這篇文章主要介紹了在Django中進行用戶注冊和郵箱驗證的方法的相關(guān)資料,需要的朋友可以參考下2016-05-05
這篇文章主要介紹了python3 打印輸出字典中特定的某個key的方法,涉及Python字典的遍歷、判斷、輸出等相關(guān)操作技巧,需要的朋友可以參考下2019-07-07
餅圖,或稱餅狀圖,是一個劃分為幾個扇形的圓形統(tǒng)計圖表,這篇文章主要為大家詳細(xì)介紹了如何使用Matplotlib繪制餅圖,需要的小伙伴可以參考下2023-08-08
python 如何在list中找Topk的數(shù)值和索引
這篇文章主要介紹了python 如何在list中找Topk的數(shù)值和索引的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
這篇文章主要介紹了Python closure閉包解釋及其注意點詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
Python實現(xiàn)日志實時監(jiān)測的示例詳解
觀察者模式：是一種行為型設(shè)計模式。主要關(guān)注的是對象的責(zé)任，允許你定義一種訂閱機制，可在對象事件發(fā)生時通知多個"觀察"該對象的其他對象。本文將利用觀察者模式實現(xiàn)日志實時監(jiān)測，需要的可以參考一下2022-04-04
pandas實現(xiàn)按照多列排序-ascending
這篇文章主要介紹了pandas實現(xiàn)按照多列排序-ascending，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
緩存是一種用于提高應(yīng)用程序性能的技術(shù),它通過臨時存儲程序獲得的結(jié)果,以便在以后需要時重用它們,本文將學(xué)習(xí)Python中的不同緩存技術(shù),感興趣的可以了解下2024-11-11
這篇文章主要介紹了20個實用的Python?Excel自動化腳本,用于自動化處理和分析Excel文件,提高工作效率,從批量填充單元格、設(shè)置行高與列寬到生成數(shù)據(jù)透視表和圖表,涵蓋了數(shù)據(jù)處理和分析的各個方面,需要的朋友可以參考下2025-02-02