save
api_attr
声明式编程模式(静态图)
该接口将传入的参数、优化器信息和网络描述保存到 model_path
。
优化器信息包含优化器使用的所有Tensor。对于Adam优化器,包含beta1、beta2、momentum等。 所有信息将保存到后缀为 .pdopt
的文件中。(如果优化器没有需要保存的Tensor(如sgd),则不会生成)。
网络描述是程序的描述。它只用于部署。描述将保存到后缀为 的文件中。
参数:
返回: 无
本文档使用 BookStack 构建