vitis_quantize.VitisQuantizer.dump_model - 3.5 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 English

This function dumps the simulation results of the quantized model, including weights and activation results:


vitis_quantize.VitisQuantizer.dump_model(
    model,
    dataset=None,
    output_dir='./dump_results',
    dump_float=False,
    weights_only=False)

Arguments

model
A tf.keras.Model object. The quantized model to dump.
dataset
A tf.data.Dataset, keras.utils.Sequence or np.numpy object. The dataset is used to dump and is not needed if weights_only is set to True.
output_dir
A string object. The directory to save the dump results.
weights_only
A bool object. Set to True only to dump the weights. Setting it to False also dumps the activation results.