pytorch-pruning

Pruning tool

This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference.

PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference

GitHub

877 stars
22 watching
202 forks
Language: Python
last commit: over 5 years ago
deep-learningpruningpytorch

Related projects:

Repository Description Stars
alecwangcq/eigendamage-pytorch A deep learning project implementing structured pruning algorithms in PyTorch for efficient neural network training and inference. 112
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,669
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 572
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,044
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
prabhuomkar/pytorch-cpp A C++ implementation of PyTorch tutorials 1,978
zsef123/efficientnets-pytorch A PyTorch implementation of EfficientNet for computer vision tasks 309
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
akanimax/pro_gan_pytorch Implementation of a deep learning model for generating high-quality images with improved stability and variation. 538
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,093
randl/shufflenetv2-pytorch An implementation of a lightweight convolutional neural network architecture for mobile devices 191
thuyngch/human-segmentation-pytorch An open source implementation of deep learning-based human segmentation models using PyTorch 558
ptrblck/prog_gans_pytorch_inference Reproducible PyTorch inference of the Progressive Growing of GANs model using CelebA training snapshot 322
jhjacobsen/pytorch-i-revnet Deep invertible neural network implementation using PyTorch for image recognition and reconstruction tasks. 390
andrewliao11/dni.pytorch An implementation of synthetic gradients to decouple neural network layers and enable scalable communication between them 119