vai_p_tensorflow の実行には、次の引数を使用できます。
引数 | タイプ | 動作 | デフォルト | 説明 |
---|---|---|---|---|
action | 文字列 | – | "" | 実行するアクション。有効なアクションは、「ana」、「prune」、「transform」、および「flops」です。 |
workspace | 文字列 | [‘ana’, ‘prune’] | "" | 出力ファイルを格納するディレクトリ。 |
input_graph | 文字列 | [‘ana’, ‘prune’, ‘flops’] | "" | ネットワークのアーキテクチャを定義した GraphDef protobuf ファイルのパス。 |
input_ckpt | 文字列 | [‘ana’, ‘prune’, ‘transform’] | "" | チェックポイント ファイルのパス。チェックポイント用に作成されるファイル名の接頭辞となります。 |
eval_fn_path | 文字列 | [‘ana’] | "" | モデルの評価に使用される Python ファイルのパス。 |
target | 文字列 | [‘ana’] | "" | モデルの精度を示す出力ノード名。 |
max_num_batches | int | [‘ana’] | なし | 評価するバッチの最大数。デフォルトでは、すべてのバッチを使用します。 |
output_graph | 文字列 | [‘prune’] | "" | プルーニング済みネットワークを格納する GraphDef protobuf ファイルのパス。 |
output_ckpt | 文字列 | [‘prune’, ‘transform’’] | "" | 重みを格納するチェックポイント ファイルのパス。 |
gpu | 文字列 | [‘ana’] | "" | 使用する GPU のデバイス ID (カンマ区切り)。 |
sparsity | float | [‘prune’] | なし | プルーニング後のネットワークの目標スパース度。 |
exclude | リピート | [‘ana’, ‘prune’] | なし | プルーニングから除外されるたたみ込みノード。 |
input_nodes | リピート | [‘flops’] | なし | 推論グラフの入力ノード。 |
input_node_shapes | リピート | [‘flops’] | なし | 入力ノードの形状。 |
output_nodes | リピート | [‘ana’, ‘prune’, ‘flops’] | なし | 推論グラフの出力ノード。 |
channel_batch | int | [‘prune’] | 2 | プルーニング後、出力チャネルの数はこの値の倍数値となります。 |