この関数は、量子化されたモデルのシミュレーション結果をダンプします。これには、重みとアクティベーションの結果が含まれます。
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オブジェクト。dataset はダンプに使用されます。weights_only をTrueに設定した場合は不要です。 - output_dir
-
stringオブジェクト。ダンプ結果を保存するディレクトリ。 - weights_only
-
boolオブジェクト。Trueに設定すると、重みのみがダンプされます。Falseに設定すると、アクティベーションの結果もダンプされます。