save_params
API属性:声明式编程(静态图)专用API
该OP从 main_program
中取出所有参数,然后将它们保存到 dirname
目录下或名为 的文件中。
注意:
- 有些变量不是参数,如学习率,全局训练步数(global step)等,但它们对于训练却是必要的。因此,调用 save_params 和 来保存和加载参数对于断点训练是不够的,这种情况下可以使用 save_persistables 和 来保存和加载训练过程中的检查点(checkpoint)。
- 如果您想要储存您的模型用于预测,请使用 save_inference_model 。更多细节请参考