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均为有效取值。
返回值说明
无。
调用示例
1 2 3 4 5 6 7 8 9 10 11 |
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") |
父主题: 公共接口