今天小编就为大家分享一篇pytorch 获取层权重,对特定层注入hook, 提取中间层输出的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如下所示： #获取模型权重 for k, v in model_2.state_dict().iteritems(): print("Layer {}".format(k)) print(v) #获取模型权重 for layer in model_2.modules(): if isinstance(layer, nn.Linear): print(layer.weight) #将一个模型权重载入另一个模型 model = VGG(make_layers(cfg['E']),