set_logging_level

功能说明

设置信息输出的级别，其中信息包括打印在屏幕上的信息以及保存到amct_log/amct_tensorflow.log文件中的信息。

函数原型

set_logging_level(print_level='info', save_level='info')

参数说明

参数名

输入/返回值

含义

使用限制

print_level

输入

控制打印信息的级别。

debug：输出DEBUG/INFO/WARNING/ERROR级别的运行信息。
info：输出INFO/WARNING/ERROR级别的运行信息。
warning：输出WARNING/ERROR级别的运行信息。
error：输出ERROR级别的运行信息。

默认值为：info

数据类型：string

save_level

输入

控制quant_info.log中保存信息的级别。

debug：输出DEBUG/INFO/WARNING/ERROR级别的运行信息。
info：输出INFO/WARNING/ERROR级别的运行信息。
warning：输出WARNING/ERROR级别的运行信息。
error：输出ERROR级别的运行信息。

默认值为：info

数据类型：string

信息级别	信息描述
DEBUG	详细的量化处理信息，包含量化因子（scale/offset）以及相关调试信息。
INFO	概要的量化处理信息，包含量化的层名、bn融合等信息。
WARNING	量化处理过程中的警告信息。
ERROR	量化处理过程中的错误信息。

信息级别不区分大小写，即Info、info、INFO均为有效取值。

返回值说明

无。

调用示例

import amct_tensorflow as amct
amct.set_logging_level(print_level="info", save_level="info")
amct.quantize_model(
     graph=tf.get_default_graph(),
     config_file="./configs/config.json",
     record_file="./record_scale_offset.txt")
   ...
amct.save_model(pb_model="./user_model.pb",
                outputs=["model/outputs"],
                record_file="./record_scale_offset.txt",
                save_path="./inference/model")

父主题： 公共接口