1. 程式人生 > 程式設計 >pytorch 實現檢視網路中的引數

pytorch 實現檢視網路中的引數

可以通過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 實現檢視網路中的引數就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。