此函数可转储量化模型的仿真结果,包括权重和激活结果:
vitis_quantize.VitisQuantizer.dump_model(
model,
dataset=None,
output_dir='./dump_results',
dump_float=False,
weights_only=False)
实参
- model
-
tf.keras.Model对象。要转储的量化模型。 - dataset
-
tf.data.Dataset、keras.utils.Sequence或np.numpy对象。该数据集用于转储,如果 weights_only 设置为True,则无需使用。 - output_dir
-
string对象。表示转储结果的保存目录。 - weights_only
-
bool对象。设置为True表示仅转储权重。将其设置为False则一并转储激活结果。