model_prune_torch.py 9.75 KB