vitis_quantize.VitisQuantizer.dump_model - 2.0 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2022-01-20
Version
2.0 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 used to dump, 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 to only dump the weights, set to False will also dump the activation results.