pytorch_nndct.get_pruning_runner - 3.5 English

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 real data.

method

Either be 'iterative' or 'one_step.'