pytorch 實現(xiàn)查看網(wǎng)絡中的參數(shù)
更新時間:2020年01月06日 09:31:52 作者:小妖精Fsky
今天小編就為大家分享一篇pytorch 實現(xiàn)查看網(wǎng)絡中的參數(shù),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
可以通過model.state_dict()或者model.named_parameters()函數(shù)查看現(xiàn)在的全部可訓練參數(shù)(包括通過繼承得到的父類中的參數(shù))
可示例代碼如下:
params = list(model.named_parameters()) (name, param) = params[28] print(name) print(param.grad) print('-------------------------------------------------') (name2, param2) = params[29] print(name2) print(param2.grad) print('----------------------------------------------------') (name1, param1) = params[30] print(name1) print(param1.grad)
以上這篇pytorch 實現(xiàn)查看網(wǎng)絡中的參數(shù)就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python結合百度語音識別實現(xiàn)實時翻譯軟件的實現(xiàn)
這篇文章主要介紹了Python結合百度語音識別實現(xiàn)實時翻譯軟件的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2021-01-01Python中實現(xiàn)傳遞未知數(shù)量的函數(shù)參數(shù)
這篇文章主要介紹了Python中實現(xiàn)傳遞未知數(shù)量的函數(shù)參數(shù)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02