昇腾社区首页
中文
注册

启动方法与参数配置

Rec SDK训练脚本中开启“STAT_ON”开关,并把日志定向到固定文件,统计信息的元数据会输出到日志文件中;配置日志统计工具并启动,统计信息会自动输出到当前目录或打屏。统计工具支持实时处理日志,需手动关闭。

  1. Rec SDK训练脚本中(如run.sh)设置环境变量STAT_ON,0为关闭统计,1为开启统计。Rec SDK默认关闭统计。
  2. 维测统计脚本在mxrec仓库,使用前请根据要分析的Rec SDK日志正确配置如下参数:
  • TARGET_REC_LOG_PATH:Rec SDK产生的训练/评估日志路径。
  • RUN_MODE:统计的运行模式,目前支持的模式有HBM_DICT_NORMAL、HBM_DICT_FAAE、HBM_DICT_HOT、HBM_DICT_FAST、DDR_DICT_NORMAL、DDR_DICT_FAAE、DDR_DICT_HOT和DDR_DICT_FAST。
  • RANK_SIZE:训练脚本运行的rank size,即卡运行的数量。
  • DISPLAY_MODE:统计信息展示的方式,支持落盘以及打屏,分别是“save_log”或者“print_screen”。
  • DISPLAY_INTERVAL:统计日志的间隔,即多少步输出一次统计信息。