TUNE_BANK_PATH
功能描述
可通过此环境变量指定调优后自定义知识库的存储路径。
针对RL调优模式:
- 若不配置此环境变量,自定义知识库默认存储在ATC安装目录的atc/data/rl/<soc_version>/custom路径下。
- 若配置此环境变量,可配置为绝对路径或相对于执行Auto Tune工具所在路径的相对路径。
- 若TUNE_BANK_PATH配置的自定义路径中包含“atc/data/rl/”,则RL调优后的最优策略直接存入默认存储路径中,即:ATC安装目录的atc/data/rl/<soc_version>/custom路径下。
- TUNE_BANK_PATH为其他自定义路径的场景,RL调优后的策略存储在配置路径的<soc version>/custom目录下,调优脚本会自动在配置的TUNE_BANK_PATH路径下创建<soc version>/custom两级目录。
配置的路径需要为已存在的目录,且执行用户具有读、写、可执行权限。
例如TUNE_BANK_PATH= ./bank_rl,则RL调优后的最优策略存入./bank_rl/<soc version>/custom目录中。
针对GA调优模式:
说明:若调优时自定义了知识库路径,后续进行模型转换时,若想直接使用自定义知识库,也需要配置上此环境变量。
配置示例
使用场景 |
配置示例 |
---|---|
使用auto_tune工具进行RL模式调优时使用 |
export TUNE_BANK_PATH=/home/HwHiAiUser/custom_tune_bank |
父主题: Auto Tune