pytorch-prunes

Pruning tool

A tool for training neural networks with pruned weights and evaluating their performance.

Code for https://arxiv.org/abs/1810.04622

GitHub

139 stars
7 watching
20 forks
Language: Python
last commit: about 5 years ago

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
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,666
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
prlz77/resnext.pytorch Reproduces ResNet-V3 with PyTorch for computer vision tasks 508
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 569
ntumslab/prune A generative approach for network embedding that preserves proximity and global ranking. 45
atgambardella/pytorch-es An implementation of an optimization algorithm for training neural networks in machine learning environments. 351
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
randl/shufflenetv2-pytorch An implementation of a lightweight convolutional neural network architecture for mobile devices 191
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,520
spandan-madan/pytorch_fine_tuning_tutorial Provides guidance on fine-tuning pre-trained models for image classification tasks using PyTorch. 279
akanimax/pro_gan_pytorch Implementation of a deep learning model for generating high-quality images with improved stability and variation. 536
rwightman/pytorch-dpn-pretrained A PyTorch implementation of Dual Path Networks (DPNs) with pre-trained weights and conversion scripts. 206