提取神經網絡中參數(放入numpy矩陣中)的方法

參考:https://blog.csdn.net/leviopku/article/details/78510977 最近需要把resnet中的參數(權重,偏置)從網絡中取出來分析,網上資料很少,懟了一下午算是成功了,記錄一下。 選擇的框架是TF,因爲pytorch提出來的參數數據結構複雜,而TF可以直接變爲numpy。 思路就是先把所有參數用tf.train.NewCheckpointReade
相關文章
相關標籤/搜索