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调优模式:
  • 若不配置此环境变量,自定义知识库默认存储在ATC安装目录的atc/data/tiling/<soc_version>/custom路径下。
  • 若配置此环境变量,则GA调优后的最优策略直接存入TUNE_BANK_PATH指定的目录下。

    可配置为绝对路径或相对于执行Auto Tune工具所在路径的相对路径,配置的路径需要为已存在的目录,且执行用户具有读、写、可执行权限。

说明:若调优时自定义了知识库路径,后续进行模型转换时,若想直接使用自定义知识库,也需要配置上此环境变量。

配置示例

使用场景

配置示例

使用auto_tune工具进行RL模式调优时使用

export TUNE_BANK_PATH=/home/HwHiAiUser/custom_tune_bank