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