Releases: VainF/Torch-Pruning
Releases · VainF/Torch-Pruning
v0.2.8: Vision transformers
- Customizable metrics for pruning plan
- Non-recursive DependencyGraph
- New modules: nn.Parameters and nn.MultiheadAttention
- Fixed some bugs
Support for channel rounding, nested outputs and customized modules
v0.2.7 v0.2.7: fixing bugs in channel rounding