load_from_file

paddle.static.load_from_file ( path ) [源代码]

从指定的文件中加载内容。

参数:

  • path (str) - 要读取的文件。

返回:

  • bytes: 文件内容。

代码示例

  1. import paddle
  2. paddle.enable_static()
  3. path_prefix = "./infer_model"
  4. # 用户自定义网络,此处用 softmax 回归为例。
  5. image = paddle.static.data(name='img', shape=[None, 28, 28], dtype='float32')
  6. label = paddle.static.data(name='label', shape=[None, 1], dtype='int64')
  7. predict = paddle.static.nn.fc(image, 10, activation='softmax')
  8. loss = paddle.nn.functional.cross_entropy(predict, label)
  9. exe = paddle.static.Executor(paddle.CPUPlace())
  10. exe.run(paddle.static.default_startup_program())
  11. # 序列化参数
  12. serialized_params = paddle.static.serialize_persistables([image], [predict], exe)
  13. # 将序列化之后的参数保存到文件
  14. params_path = path_prefix + ".params"
  15. paddle.static.save_to_file(params_path, serialized_params)
  16. # 从文件加载序列化之后的参数
  17. serialized_params_copy = paddle.static.load_from_file(params_path)