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的文件前缀。格式为
目录名称/文件前缀
。如果文件前缀为空字符串,会引发异常。
- state_dict (dict of Parameters) – 要保存的模型参数的
返回: 无
代码示例
- import paddle.fluid as fluid
- with fluid.dygraph.guard():
- emb = fluid.dygraph.Embedding(
- size=[10, 32],
- param_attr='emb.w',
- is_sparse=False)
- state_dict = emb.state_dict()
- fluid.save_dygraph(state_dict, "paddle_dy") # 会保存为 paddle_dy.pdparams
- adam = fluid.optimizer.Adam(
- learning_rate=fluid.layers.noam_decay(100, 10000),
- parameter_list = emb.parameters())
- state_dict = adam.state_dict()
- fluid.save_dygraph(state_dict, "paddle_dy") # 会保存为 paddle_dy.pdopt