下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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均为有效取值。

返回值说明

无。

调用示例

 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")
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词