fluid.io.save_inference_model()含义及其参数含义
时间: 2024-05-27 18:10:27 浏览: 159
fluid_inference_install_dir_windows.zip
5星 · 资源好评率100%
fluid.io.save_inference_model() 是 PaddlePaddle 框架中模型保存函数,用于将训练好的模型保存成二进制文件,以便后续可以加载模型进行预测。该函数有三个参数:
1. dirname:保存模型的路径;
2. feeded_var_names:一个 Python list,指定需要传入模型的变量名,类型为字符串;
3. target_vars:一个 Python list,指定需要从模型中获取的变量名,类型为字符串。
阅读全文