如何将使用save_dygraph接口保存的模型使用save_inference_model接口保存
收藏
如何将使用save_dygraph接口保存的模型使用save_inference_model接口保存:
fluid.dygraph.save_dygraph(net.state_dict(), train_parameters["save_persistable_dir"])
这样保存的模型,可以使用
save_inference_model 接口保存成.pdmodel和.pdiparam格式的模型吗?
0
收藏
请登录后评论
可以参考官网文档将动态图转换成静态图,然后使用静态图save_inference_model接口保存模型
参考链接:
https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/04_dygraph_to_static/basic_usage_cn.html