set_logging_level
功能说明
设置信息输出的级别，其中信息包括打印在屏幕上的信息以及保存到amct_log/amct_tensorflow.log文件中的信息。
函数原型
set_logging_level(print_level='info', save_level='info')
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
|---|---|---|---|
print_level |
输入 |
控制打印信息的级别。
|
默认值为：info 数据类型：string |
save_level |
输入 |
控制quant_info.log中保存信息的级别。
|
默认值为：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")
父主题： 公共接口
