pytorch 實現檢視網路中的引數
阿新 • • 發佈:2020-01-09
可以通過model.state_dict()或者model.named_parameters()函式檢視現在的全部可訓練引數(包括通過繼承得到的父類中的引數)
可示例程式碼如下:
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 實現檢視網路中的引數就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。