save_dygraph

注意：该API仅支持【动态图】模式

  • paddle.fluid.dygraph.save_dygraph(state_dict, model_path)[源代码]

该接口将传入的参数或优化器的 dict 保存到磁盘上。

state_dict 是通过 Layer 的 state_dict() 方法得到的。

注： model_path 不可以是一个目录。

该接口会根据 state_dict 的内容，自动给 model_path 添加 .pdparams 或者 .pdopt 后缀， 生成 model_path + ".pdparams" 或者 model_path + ".pdopt" 文件。

  • 参数:
    • state_dict (dict of Parameters) – 要保存的模型参数的 dict 。
    • model_path (str) – 保存state_dict的文件前缀。格式为 目录名称/文件前缀。如果文件前缀为空字符串，会引发异常。

返回: 无

代码示例

  1. import paddle.fluid as fluid
  2.  
  3. with fluid.dygraph.guard():
  4.     emb = fluid.dygraph.Embedding(
  5.         size=[10, 32],
  6.         param_attr='emb.w',
  7.         is_sparse=False)
  8.     state_dict = emb.state_dict()
  9.     fluid.save_dygraph(state_dict, "paddle_dy")  # 会保存为 paddle_dy.pdparams
  10.  
  11.     adam = fluid.optimizer.Adam(
  12.         learning_rate=fluid.layers.noam_decay(100, 10000),
  13.         parameter_list = emb.parameters())
  14.     state_dict = adam.state_dict()
  15.     fluid.save_dygraph(state_dict, "paddle_dy")  # 会保存为 paddle_dy.pdopt