pytorch_nndct.get_pruning_runner - 2.5 Japanese

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

Document ID
UG1333
Release Date
2022-06-15
Version
2.5 Japanese

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

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