概览

使用amct_tensorflow命令行进行量化时,命令有两种方式,用户根据实际情况进行选择:
  1. amct_tensorflow calibration param1=value1 param2=value2 ...(value值前面不能有空格,否则会导致截断,param取的value值为空)
  2. amct_tensorflow calibration param1 value1 param2 value2 ...

参数项

参数说明

训练后量化场景

QAT模型适配Ascend模型场景

--help或--h

打印帮助信息。

可选

可选

--model

待量化网络的模型文件与路径,格式为.pb。

必填

必填

--save_path

量化后模型的存放路径。

必填

必填

--outputs

指定原始模型输出Tensor的名称。

必填

必填

--input_shape

指定模型输入数据的shape。

若未使用--evaluator参数,则该参数必填。

可选

不适用

--data_dir

与模型匹配的bin格式数据集路径。

若未使用--evaluator参数,则该参数必填。

可选

不适用

--data_types

输入数据的类型。

若未使用--evaluator参数,则该参数必填。

可选

不适用

--evaluator

基于"Evaluator"基类并且包含evaluator评估器的python脚本。

该参数与--input_shape--data_dir--data_types不能同时使用。

可选

不适用

--calibration_config

训练后量化简易配置文件路径与文件名。

可选

不适用

--batch_num

训练后量化推理阶段的batch数。

可选

不适用

--feature

模型压缩特性使能开关。

可选

不适用