This API has the following methods:
get_pruning_runner(model, inputs, method)- model
- A torch.nn.Module object to prune.
- inputs
- 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 the real data.
- method
- Either be 'iterative' or 'one_step'.