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