Created by: Shixiaowei02
[cherry-pick from #22562 ]
Python 接口支持从内存存取预测模型,以满足加解密需求。
load_inference_model(dirname, executor, model_filename=None, params_filename=None, pserver_endpoints=None)
当 dirname
为 None
时,可向 model_filename
和 params_filename
分别传递包含模型和参数的字符串,实现模型内存读取。
save_persistables(executor, dirname, main_program=None, filename=None)
当 dirname
和 filename
为 None
时,save_persistables 将返回参数字符串,而不再将模型文件存盘。
上述改动涉及到 save_vars
、save_params
的返回值变化,和 load_vars
的参数含义的增加。本改动是向后兼容的。