AMCT(PyTorch)
AMCT量化过程中的日志信息以及日志级别,可以通过环境变量设置,本章节给出详细设置方法。
日志包括打印在屏幕上的日志以及保存到amct_log/amct_pytorch.log文件中的日志,该部分环境变量为可选配置,如果不设置,则按照默认日志级别,默认级别为INFO。
amct_pytorch.log文件中的日志重新执行量化时会被覆盖,请用户自行进行保存。此外,由于生成的日志文件大小和所要量化模型层数有关,请用户确保AMCT工具所在服务器有足够空间:
以量化resnet101模型为例,日志级别设置为INFO,日志文件大小为5KB左右,中间临时文件大小为170MB左右;日志级别设置为DEBUG,日志文件大小为2MB左右,中间临时文件大小为170MB左右。
- 变量取值
日志打印级别通过如下两个变量设置:
- AMCT_LOG_FILE_LEVEL: 控制amct_pytorch.log日志文件的信息级别以及生成精度仿真模型时,对应量化层生成的日志文件信息级别。
- AMCT_LOG_LEVEL:控制屏幕输出的信息级别。
有效取值以及含义如表1所示。信息级别不区分大小写,即Info、info、INFO均为有效取值。
- 使用示例
如下命令只是样例,用户根据实际情况进行设置。
- 将量化日志amct_pytorch.log信息级别设置为INFO级别。
export AMCT_LOG_FILE_LEVEL=INFO
- 将屏幕打印输出信息级别设置为INFO级别。
export AMCT_LOG_LEVEL=INFO
- 将量化日志amct_pytorch.log信息级别设置为INFO级别。
父主题: 安装后处理