pytorch_nndct.get_pruning_runner - 3.0 日本語

Vitis AI オプティマイザー ユーザー ガイド (UG1333)

Document ID
UG1333
Release Date
2023-01-12
Version
3.0 日本語

この API には次のメソッドがあります。

get_pruning_runner(model, inputs, method)
model
プルーニングする torch.nn.Module オブジェクト。
inputs
モデル推論用の入力として使用される単一の torch.Tensor または torch.Tensor のリスト。実際のデータは必要ありません。形状とデータ型が実際のデータと同じであれば、無作為に生成されるテンソルでもかまいません。
method
'iterative' または 'one_step' のいずれかです。