This API has the following methods:
get_pruning_runner(model, inputs, method)
torch.nn.Moduleobject to prune.
- A single or a list of torch. Tensor used as inputs for model inference. It does not need real data. It can be a randomly generated tensor of the same shape and data type as real data.
- Either be 'iterative' or 'one_step.'