model_prune_torch.py 9.81 KB