vitis_quantize.VitisQuantizer - 1.4.1 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-12-13
Version
1.4.1 English

The construction function of class VitisQuantizer.


vitis_quantize.VitisQuantizer(
    float_model, 
    quantize_strategy='8bit', 
    custom_quantize_strategy=None, 
    custom_objects={})

Arguments

model
A tf.keras.Model object, containing the configurations for quantization.
quantize_strategy
A string object of the quantize strategy type. Available values are 8bit and 8bit_tqt. 8bit is the default strategy that uses the Straight-Through-Estimator. 8bit_tqt is a new strategy introduced in Vitis AI 1.4 which uses Trained-Threshold in quantizers and may better results for QAT.
Note: 8bit_tqt strategy should only be used in QAT and be used together with init_quant=True to get the best performance.
custom_quantize_strategy
A string object, the file path of custom quantize strategy JSON file.
custom_objects
A Dict object, mapping names (strings) to custom classes or functions.