pytorch/torch/ao/pruning
LINGAO XIAO e7b2430818 add pruning method: Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (#95689)
add `class FPGMStructured`
add `function FPGM_structured()`
add `function _validate_distance_type()`
add `function _compute_distance()`

Implement method mentioned in issue #39765

---
FPGMSparsifier Implement with the new pytorch pruning API torch.ao.pruning.
It is a structured pruning method, and it is added under torch.ao.pruning._experimental. Test cases are added at `test_structured_sparsifier.py`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/95689
Approved by: https://github.com/jcaip
2023-08-02 16:24:42 +00:00
..
_experimental add pruning method: Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration (#95689) 2023-08-02 16:24:42 +00:00
scheduler [BE] Enable ruff's UP rules and autoformat ao/ (#105430) 2023-07-19 13:44:37 +00:00
sparsifier Format: fixing multiple string concatenation in single line (#106013) 2023-07-26 18:39:18 +00:00
__init__.py Back out "[core][pruning][be] rename BaseSparsifier to BasePruner (#98747)" (#99171) 2023-04-15 00:37:45 +00:00
_mappings.py [BE] Update flake8-comprehensions and adapt to rule C418 (#99178) 2023-04-15 15:33:42 +00:00