vitis_quantize.VitisQuantizer.dump_model - 3.0 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-02-24
Version
3.0 简体中文

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


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 将一并转储激活结果。