EigenDamage-Pytorch

Pruning algorithm

A deep learning project implementing structured pruning algorithms in PyTorch for efficient neural network training and inference.

Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934

GitHub

112 stars
5 watching
19 forks
Language: Python
last commit: over 6 years ago
deep-learningfisher-matrixmachine-learningnetwork-pruningpytorch

Related projects:

RepositoryDescriptionStars
jacobgil/pytorch-pruningThis project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference.877
zhanghang1989/pytorch-encodingA Python framework for building deep learning models with optimized encoding layers and batch normalization.2,044
bayeswatch/pytorch-prunesA tool for training neural networks with pruned weights and evaluating their performance.140
javeywang/pyramid-attention-networks-pytorchAn implementation of a deep learning model using PyTorch for semantic segmentation tasks.237
kaiyangzhou/dassl.pytorchA PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.1,236
rguthrie3/deeplearningfornlpinpytorchA comprehensive tutorial on deep learning for natural language processing with PyTorch, covering the basics and advancing to linguistic structure prediction.1,942
akanimax/pro_gan_pytorchImplementation of a deep learning model for generating high-quality images with improved stability and variation.538
atgambardella/pytorch-esAn implementation of an optimization algorithm for training neural networks in machine learning environments.351
andrewliao11/dni.pytorchAn implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them119
graal-research/poutyneA PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities572
rentruewang/koilaA lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution1,823
thuyngch/human-segmentation-pytorchAn open source implementation of deep learning-based human segmentation models using PyTorch558
dgriff777/rl_a3c_pytorchAn implementation of an A3C algorithm for reinforcement learning in Pytorch, with various optimizations and extensions to accelerate training.562
onlytailei/a3c-pytorchAn implementation of Advantage async Actor-Critic Algorithms in PyTorch for Deep Reinforcement Learning114
ikostrikov/pytorch-trpoA PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks435