概览
本节给出命令行场景所使用的参数概览。
- 如果通过amct_tensorflow calibration --help命令查询出的参数未解释在表1,则说明该参数预留或适用于其他芯片版本,用户无需关注。
- 使用amct_tensorflow命令行进行量化时,命令有两种方式,用户根据实际情况进行选择:
- amct_tensorflow calibration param1=value1 param2=value2 ...(value值前面不能有空格,否则会导致截断,param取的value值为空)
- amct_tensorflow calibration param1 value1 param2 value2 ...
参数项 |
参数说明 |
训练后量化场景 |
QAT模型适配Ascend模型场景 |
---|---|---|---|
打印帮助信息。 |
可选 |
可选 |
|
待量化网络的模型文件与路径,格式为.pb。 |
必填 |
必填 |
|
量化后模型的存放路径。 |
必填 |
必填 |
|
指定原始模型输出Tensor的名称。 |
必填 |
必填 |
|
指定模型输入数据的shape。 若未使用--evaluator参数,则该参数必填。 |
可选 |
不适用 |
|
与模型匹配的bin格式数据集路径。 若未使用--evaluator参数,则该参数必填。 |
可选 |
不适用 |
|
输入数据的类型。 若未使用--evaluator参数,则该参数必填。 |
可选 |
不适用 |
|
基于"Evaluator"基类并且包含evaluator评估器的python脚本。 该参数与--input_shape、--data_dir、--data_types不能同时使用。 |
可选 |
不适用 |
|
训练后量化简易配置文件路径与文件名。 |
可选 |
不适用 |
|
训练后量化推理阶段的batch数。 |
可选 |
不适用 |
|
模型压缩特性使能开关。 |
可选 |
不适用 |
父主题: 命令行参数说明