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')