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
111 stars
5 watching
19 forks
Language: Python
last commit: over 4 years ago deep-learningfisher-matrixmachine-learningnetwork-pruningpytorch
Related projects:
Repository | Description | Stars |
---|---|---|
jacobgil/pytorch-pruning | This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. | 875 |
zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,041 |
bayeswatch/pytorch-prunes | A tool for training neural networks with pruned weights and evaluating their performance. | 139 |
javeywang/pyramid-attention-networks-pytorch | An implementation of a deep learning model using PyTorch for semantic segmentation tasks. | 235 |
kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
rguthrie3/deeplearningfornlpinpytorch | A comprehensive tutorial on deep learning for natural language processing with PyTorch, covering the basics and advancing to linguistic structure prediction. | 1,940 |
akanimax/pro_gan_pytorch | Implementation of a deep learning model for generating high-quality images with improved stability and variation. | 536 |
atgambardella/pytorch-es | An implementation of an optimization algorithm for training neural networks in machine learning environments. | 350 |
andrewliao11/dni.pytorch | An implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them | 118 |
graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 569 |
rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,821 |
thuyngch/human-segmentation-pytorch | An open source implementation of deep learning-based human segmentation models using PyTorch | 558 |
dgriff777/rl_a3c_pytorch | An implementation of an A3C algorithm for reinforcement learning in Pytorch, with various optimizations and extensions to accelerate training. | 562 |
onlytailei/a3c-pytorch | An implementation of Advantage async Actor-Critic Algorithms in PyTorch for Deep Reinforcement Learning | 114 |
ikostrikov/pytorch-trpo | A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks | 433 |