--mdl_bank_path
关联参数
该参数需要与--buffer_optimize参数配合使用,通过利用高速缓存--buffer_optimize暂存数据的方式,达到提升性能的目的。
参数取值
参数值:子图调优后自定义知识库路径。
参数值格式:支持大小写字母(a-z,A-Z)、数字(0-9)、下划线(_)、中划线(-)、句点(.)。
参数默认值:${HOME}/Ascend/latest/data/aoe/custom/graph/<soc_version>
推荐配置及收益
无。
示例
例如子图调优后自定义知识库的路径为$HOME/custom_module_bank,则使用示例为:
--mdl_bank_path=$HOME/custom_module_path --buffer_optimize=l2_optimize
依赖约束
加载子图调优后自定义知识库路径优先级:--mdl_bank_path参数加载路径>TUNE_BANK_PATH环境变量设置路径>默认子图调优后自定义知识库路径。
- 如果模型转换前,通过TUNE_BANK_PATH环境变量指定了子图调优自定义知识库路径,模型转换时又通过--mdl_bank_path参数加载了自定义知识库路径,该场景下以--mdl_bank_path参数加载的路径为准,TUNE_BANK_PATH环境变量设置的路径不生效。
- --mdl_bank_path参数和环境变量指定路径都不生效或无可用自定义知识库,则使用默认自定义知识库路径。
- 如果上述路径下都无可用的自定义知识库,则atc工具会查找子图调优内置知识库,该路径为:${INSTALL_DIR}/compiler/data/fusion_strategy/built-in
支持的芯片型号
昇腾310B AI处理器
父主题: 模型调优选项