Call prune()
method to get a pruned model. The
ratio
is the proportion of FLOPs expected to be
reduced and output_script
specifies the path of the
Python script that can be used to rebuild the pruned model.
model = pruner.prune(ratio=0.1, output_script='pruned_resnet18.py')