この関数は、量子化されたモデルのシミュレーション結果をダンプします。これには、重みとアクティベーションの結果が含まれます。
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
に設定すると、アクティベーションの結果もダンプされます。