vitis_quantize.VitisQuantizer.dump_model - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

此函数可转储量化模型的仿真结果,包括权重和激活结果:


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.Datasetkeras.utils.Sequencenp.numpy 对象。该数据集用于转储,如果 weights_only 设置为 True,则无需使用。
output_dir
string 对象。表示转储结果的保存目录。
weights_only
bool 对象。设置为 True 表示仅转储权重。将其设置为 False 则一并转储激活结果。