Simplified Configuration File

The following table describes the simplified configuration file. Find the file in /amct_tensorflow/proto/calibration_config_ascend_tf.proto under the AMCT installation directory.

Table 1 calibration_config_ascend_tf.proto

Parameter

Required/Optional

Type

Field

Description

AMCTConfig

-

-

-

Simplified PTQ configuration of AMCT.

Optional

Boolean

activation_offset

Whether to quantize activations with offset. It is a global configuration parameter.

  • true: with offset. Activations are asymmetrically quantized.
  • false: without offset. Activations are symmetrically quantized.

Repeated

String

skip_layers

Layers to skip quantization.

Repeated

String

skip_layer_types

Types of layers to skip quantization.

Optional

CalibrationConfig

common_config

Common quantization configuration, which is a global parameter. Use this configuration if a layer is not overridden by override_layer_types or override_layer_configs.

Parameter priority: override_layer_configs > override_layer_types > common_config

Repeated

OverrideLayerType

override_layer_types

Certain types of layers to override the quantization configurations. It is used to determine which layers are to be differentiatedly quantized.

By using this parameter, you can perform differentiated quantization on some layers to change the quantization factor search step from 0.01 to 0.02.

Parameter priority: override_layer_configs > override_layer_types > common_config

Repeated

OverrideLayer

override_layer_configs

Layer to override the quantization configurations. It is used to determine which layers are to be differentiatedly quantized.

By using this parameter, you can perform differentiated quantization on some layers to change the quantization factor search step from 0.01 to 0.02.

Parameter priority: override_layer_configs > override_layer_types > common_config

Optional

Boolean

do_fusion

BN fusion switch. Defaults to true, indicating BN fusion enabled.

Repeated

String

skip_fusion_layers

Layers to skip BN fusion.

OverrideLayerType

-

-

-

Quantization configuration to override by layer type.

Required

String

layer_type

Quantizable layer type.

Required

CalibrationConfig

calibration_config

Quantization configuration to override.

OverrideLayer

-

-

-

Quantization configuration to override by layer.

Required

String

layer_name

Layers to override.

Required

CalibrationConfig

calibration_config

Quantization configuration to override.

CalibrationConfig

-

-

-

Calibration-based quantization configuration.

-

ARQuantize

arq_quantize

Weight quantization algorithm configuration.

arq_quantize: ARQ algorithm configuration.

-

FMRQuantize

ifmr_quantize

Activation quantization algorithm configuration.

ifmr_quantize: IFMR quantization algorithm configuration. Currently, only the IFMR algorithm is supported.

ARQuantize

-

-

-

ARQ algorithm configuration. For details about the algorithm, see ARQ Algorithm.

Optional

Boolean

channel_wise

Whether to use different quantization factors for each channel.

FMRQuantize

-

-

-

FMR quantization algorithm configuration. For details about the algorithm, see IFMR Algorithm.

Optional

Float

search_range_start

Quantization factor search start.

Optional

Float

search_range_end

Quantization factor search end.

Optional

Float

search_step

Quantization factor search step.

Optional

Float

max_percentile

Upper bound for searching for the largest.

Optional

Float

min_percentile

Lower bound for searching for the smallest.

Optional

Boolean

asymmetric

Whether to perform symmetric quantization. It is used to select the layer-wise quantization algorithm.

  • true: asymmetric quantization
  • false: symmetric quantization

If this parameter is set for override_layer_configs, override_layer_types, and common_config, or

if the activation_offset parameter is set, the priority is as follows:

override_layer_configs > override_layer_types > common_config > activation_offset

The following is an example of the simplified configuration file (quant.cfg):

# global quantize parameter
activation_offset : true
skip_layers : "Opname"
skip_layer_types:"Optype"
do_fusion: true
skip_fusion_layers : "Opname"
common_config : {
    arq_quantize : {
        channel_wise : true
    }
    ifmr_quantize : {
        search_range_start : 0.7
        search_range_end : 1.3
        search_step : 0.01
        max_percentile : 0.999999
        min_percentile : 0.999999
        asymmetric : true
    }
}
 
override_layer_types : {
    layer_type : "Optype"
    calibration_config : {
        arq_quantize : {
            channel_wise : false
        }
        ifmr_quantize : {
            search_range_start : 0.8
            search_range_end : 1.2
            search_step : 0.02
            max_percentile : 0.999999
            min_percentile : 0.999999
            asymmetric : true
        }
    }
}
 
override_layer_configs : {
    layer_name : "Opname"
    calibration_config : {
        arq_quantize : {
            channel_wise : true
        }
        ifmr_quantize : {
            search_range_start : 0.8
            search_range_end : 1.2
            search_step : 0.02
            max_percentile : 0.999999
            min_percentile : 0.999999
            asymmetric : true
        }
    }
}