TUNE_BANK_PATH
功能描述
可通过此环境变量指定调优后自定义知识库的存储路径。
设置的存储路径必须为绝对路径或相对于执行AOE调优引擎所在路径的相对路径,配置的路径需要为已存在的目录,且执行用户具有读、写、可执行权限。若配置的TUNE_BANK_PATH路径不存在或用户无权限,则调优进程会报错并退出。
- 子图自定义知识库
- 若不配置此环境变量,自定义知识库默认存储在:${HOME}/Ascend/latest/data/aoe/custom/graph/${soc_version}路径下。
- 若配置了此环境变量,自定义知识库存储在该环境变量路径下。
- 算子自定义知识库
- 若不配置此环境变量,自定义知识库默认存储在:${HOME}/Ascend/latest/data/aoe/custom/op/${soc_version}路径下。
- 若配置此环境变量,则调优后的最优策略存储在配置路径的${soc_version}目录下。
在多用户共享知识库场景下,共享知识库的用户需要设置TUNE_BANK_PATH为同一路径,并且对配置的路径具有读、写权限。
若调优时自定义了知识库路径,后续进行模型转换时,若想直接使用自定义知识库,也需要配置此环境变量。
配置示例
export TUNE_BANK_PATH=/home/HwHiAiUser/custom_tune_bank
是否必选
否
使用约束
无
支持的型号
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas 推理系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200I/500 A2推理产品
父主题: AOE调优