此函数可转储量化模型的仿真结果,包括权重和激活结果。
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
将一并转储激活结果。