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
ornp.numpy
object. The dataset is used to dump and is not needed if weights_only is set toTrue
. - output_dir
- A
string
object. The directory to save the dump results. - weights_only
- A
bool
object. Set toTrue
only to dump the weights. Setting it toFalse
also dumps the activation results.