save

paddle.static. save ( program, model_path, pickle_protocol=2 ) [source]
Api_attr

Static Graph

This function save parameters, optimizer information and network description to model_path.

The parameters contains all the trainable Tensor, will save to a file with suffix “.pdparams”. The optimizer information contains all the Tensor used by optimizer. For Adam optimizer, contains beta1, beta2, momentum etc. All the information will save to a file with suffix “.pdopt”. (If the optimizer have no Tensor need to save (like SGD), the fill will not generated). The network description is the description of the program. It’s only used for deployment. The description will save to a file with a suffix “.pdmodel”.

Parameters
  • program (Program) – The program to saved.

  • model_path (str) – the file prefix to save the program. The format is “dirname/file_prefix”. If file_prefix is empty str. A exception will be raised

  • pickle_protocol (int, optional) – The protocol version of pickle module must be greater than 1 and less than 5. Default: 2

Returns

None

Examples

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")