Torch-Pruning

Torch-Pruning copied to clipboard

Thanks for the excellent work! But in general, it need the definition of model class for deploying. Only get the model weights and architecture via torch.save(model, save_path) in other machines may get some trouble (for examples). Is there some ways to get the pruning model defination? Thanks :D

Hi @nick-zoo , Any progress on it? I am also interested in this.