昇腾社区首页
中文
注册

整体约束和接口列表

整体约束

  • 若接口中存在需要用户输入文件路径的参数,请确保输入路径正确,AMCT不会对路径做安全校验。
  • 若接口中存在需要用户输入文件路径的参数,重新执行量化时,该参数相关取值将会被覆盖;量化打屏日志中也会有相关文件被覆盖的warning风险提示信息。

接口列表

分类

接口名称

功能描述

公共接口

set_logging_level

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

训练后量化接口

create_quant_config_ascend

训练后量化接口,根据图的结构找到所有可量化的层,自动生成量化配置文件,并将可量化层的量化配置信息写入文件。

quantize_model_ascend

训练后量化接口,将输入的待量化的图结构按照给定的量化配置文件进行量化处理,在传入的图结构中插入量化相关的算子,生成量化因子记录文件record_file,并返回量化处理新增的算子列表。

save_model_ascend

训练后量化接口,根据量化因子记录文件record_file将用户待量化的原始pb模型保存为可以在NPU(昇腾AI处理器)环境进行在线推理的pb模型文件。