save¶

paddle.static. save ( program, model_path, protocol=4, **configs ) [源代码] ¶

将传入的参数、优化器信息和网络描述保存到 model_path 。

参数包含所有的可训练 Variable，将保存到后缀为 .pdparams 的文件中。

优化器信息包含优化器使用的所有 Tensor。对于 Adam 优化器，包含 beta1、beta2、momentum 等。 所有信息将保存到后缀为 .pdopt 的文件中。（如果优化器没有需要保存的 Tensor（如 sgd），则不会生成）。

网络描述是程序的描述。它只用于部署。描述将保存到后缀为 .pdmodel 的文件中。

import paddle
import paddle.static as static

paddle.enable_static()

x = static.data(name="x", shape=[10, 10], dtype='float32')
y = static.nn.fc(x, 10)
z = static.nn.fc(y, 10)

place = paddle.CPUPlace()
exe = static.Executor(place)
exe.run(static.default_startup_program())
prog = static.default_main_program()

static.save(prog, "./temp")