1 个仓库
Awesome GitHub RepositoriesPyTorch Pruning Frameworks
A framework for tracing dependencies and structurally pruning channels in deep learning models to reduce size and computation.
Distinct from Model Pruning: Distinct from Model Pruning: specifically targets PyTorch framework integration with dependency graph tracing and structural channel pruning capabilities.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · PyTorch Pruning Frameworks. Refine with filters or upvote what's useful.
Awesome PyTorch Pruning Frameworks GitHub Repositories
用 AI 发现最棒的仓库。我们将通过 AI 为您搜索最匹配的仓库。
- vainf/torch-pruning
VainF/Torch-Pruning
3,319在 GitHub 上查看Torch-Pruning is a deep learning model pruning tool and neural network optimization toolkit designed for PyTorch. The library analyzes neural network graphs to remove redundant parameters and channels, reducing model size and computational costs. The framework traces computational graphs dynamically using sample inputs to map relational dependencies and identify coupled layers that require simultaneous pruning. It evaluates and ranks layer parameters across the entire network using global importance ranking and applies structural constraints to optimize overall outcomes. Additional capabili
A framework for tracing dependencies and structurally pruning channels in deep learning models to reduce size and computation.
Pythonefficient-deep-learningllmmodel-compression
