プルーナーには、次の 2 つの引数が必要です。
- プルーニングするモデル
- モデルによる推論に必要な入力
import torch
from pytorch_nndct import OFAPruner
inputs = torch.randn([1, 3, 224, 224], dtype=torch.float32)
pruner = OFAPruner(model, inputs)
注記: 入力は実際のデータである必要はありません。実際のデータと同じ形状と型を持つデータであれば、無作為に生成されるダミー データでもかまいません。