在pytorch中查看可訓練參數(shù)的例子
更新時間：2019年08月18日 10:56:50 作者：玩人
今天小編就為大家分享一篇在pytorch中查看可訓練參數(shù)的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
pytorch中我們有時候可能需要設定某些變量是參與訓練的，這時候就需要查看哪些是可訓練參數(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)
這樣打印出的結果就是模型中所有的可訓練參數(shù)列表！
以上這篇在pytorch中查看可訓練參數(shù)的例子就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python實現(xiàn)基于蒙特卡洛算法過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了python將圖片文件轉換成base64編碼的方法,涉及Python操作base64編碼的技巧,需要的朋友可以參考下2015-03-03
這篇文章主要給大家介紹了關于python中remove函數(shù)的踩坑記錄，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-01-01
這篇文章主要介紹了Python 常見加密操作的實現(xiàn)，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-03-03
這篇文章主要介紹了python 層次聚類算法圖文示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-08-08
今天小編就為大家分享一篇關于Python中shapefile轉換geojson的示例，小編覺得內容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-01-01
這篇文章主要介紹了用python給自己做一款小說閱讀器過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Python 中單例模式的實現(xiàn)方法，文章圍繞主題展開詳細的內容介紹，具有一定的參考價值，感興趣的小伙伴可以學習一下下面文章詳細內容2022-08-08
這篇文章主要介紹了教你如何快速上手Selenium，文中附含詳細示例代碼，包含基本使用以及操作步驟，有需要的朋友可以參考下，希望大家多多討論交流2021-08-08
這篇文章主要為大家詳細介紹了如何基于Python實現(xiàn)簡單的溫度轉換程序,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2023-10-10