この関数は、浮動小数点モデルの検査を実行します。
VitisInspector.inspect_model(model,
input_shape=None,
dump_model=True,
dump_model_file="inspect_model.h5",
plot=True,
plot_file="model.svg",
dump_results=True,
dump_results_file="inspect_results.txt",
verbose=0)
引数
- model
- tf.keras.Model インスタンス。検査する浮動小数点モデルを指定します。浮動小数点モデルには具体的な入力形状が必要です。具体的な入力形状を指定して浮動小数点モデルを構築するか、または input_shape 引数を指定して inspect_model を呼び出してください。
- input_shape
- list(int) または list(list(int))。各入力レイヤーの入力形状が含まれます。設定しない場合、入力レイヤー内のデフォルト形状情報が使用されます。複数の入力がある場合、たとえば inspect_model(model, input_shape=[224, 224, 3]) または inspect_model(model, input_shape=[[224, 224, 3], [64, 1]]) のように、形状のリストを使用します。すべての次元に具体的な値が必要です。次元 batch_size は省略されます。デフォルト値は None です。
- dump_model
- ブール型。検査済みモデルをダンプしてモデルをディスクに保存するかどうかを指定します。デフォルト値は False です。
- dump_model_file
- 文字列。検査済みモデル ファイルのパス。デフォルト値は inspect_model.h5 です。
- plot
-
ブール型。モデルの検査結果を
graphviz
によってプロットし、イメージをディスクに保存するかどうかを指定します。モデルの検査結果を変更のヒントと共に可視化する場合に役に立ちます。ヒントを表示できるのは、.svg
など一部のタイプの出力ファイルのみです。デフォルト値は False です。 - plot_file
- 文字列。モデルをプロットする場合にモデル イメージ ファイルのファイル パスを指定します。デフォルト値は model. svg です。
- dump_results
- ブール型。検査結果をダンプしてテキストをディスクに保存するかどうかを指定します。画面に表示されるログよりも詳細なレイヤーごとの結果が、テキスト ファイルにダンプされます。デフォルト値は False です。
- dump_results_file
- 文字列。検査結果のテキスト ファイルのファイル パス。デフォルト値は inspect_results.txt です。
- verbose
- int。ログの冗長レベル。verbose の値が大きいほど、詳細な結果ログが表示されます。デフォルト値は 0 です。