提取神经网络中参数(放入numpy矩阵中)的方法

参考:https://blog.csdn.net/leviopku/article/details/78510977 最近需要把resnet中的参数(权重,偏置)从网络中取出来分析,网上资料很少,怼了一下午算是成功了,记录一下。 选择的框架是TF,因为pytorch提出来的参数数据结构复杂,而TF可以直接变为numpy。 思路就是先把所有参数用tf.train.NewCheckpointReade
相关文章
相关标签/搜索