如何导出指定层的权重，如导出最后一层的weights和bias？
收藏
如何导出指定层的权重，如导出最后一层的weights和bias？
旧鸱c_
已解决
2#
回复于2021-08
在动态图中，使用paddle.save API， 并将最后一层的layer.state_dict() 传入至save方法的obj 参数即可， 然后使用paddle.load 方法加载对应层的参数值。详细可参考API文档save 和load。 在静态图中，使用paddle.static.save_vars保存指定的vars，然后使用paddle.static.load_vars加载对应层的参数值。具体示例请见API文档：load_vars 和 save_vars 。
0
收藏
请登录后评论
在动态图中，使用paddle.save API， 并将最后一层的layer.state_dict() 传入至save方法的obj 参数即可， 然后使用paddle.load 方法加载对应层的参数值。详细可参考API文档save 和load。
在静态图中，使用paddle.static.save_vars保存指定的vars，然后使用paddle.static.load_vars加载对应层的参数值。具体示例请见API文档：load_vars 和 save_vars 。