1 repositorio
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
Encuentra los mejores repositorios con IA.Buscaremos los repositorios que mejor coincidan usando IA.
- vainf/torch-pruning
VainF/Torch-Pruning
3,319Ver en GitHubTorch-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
