tensorflow / tensorflow/model-optimization
Is it possible to implement post-training pruning?
Open
@fredrec is already working on this.
Since Mar 16, 2021.
feature request
- Dominant language
- Python
- Stars
- 1.6k
- Forks
- 349
- Avg merge
- 3d 2h
- Merged PRs (30d)
- 1
Description
MacOS
- TensorFlow version (you are using): 2.3
- Are you willing to contribute it (Yes/No): No
Motivation
The document just shows how to prune the model during training. Each time, we need to call model.fit. During the process, other weights will be modified. We just want to set some weights to zero. In the meantime, don't change other weights.
The feature have been implemented in Pytorch.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.