层权重

可以快速上手的开发文档

#层权重

pytorch 获取层权重,对特定层注入hook, 提取中间层输出的方法

如下所示:#获取模型权重fork,vinmodel_2.state_dict().iteritems():print("Layer{}".format(k))print(v)#获取模..