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
140 stars
7 watching
20 forks
Language: Python
last commit: over 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. | 877 |
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 |
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 |
prlz77/resnext.pytorch | Reproduces ResNet-V3 with PyTorch for computer vision tasks | 511 |
zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,044 |
graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 572 |
ntumslab/prune | A framework for unsupervised network embedding using a multi-task Siamese neural network | 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,523 |
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. | 538 |
rwightman/pytorch-dpn-pretrained | A PyTorch implementation of Dual Path Networks (DPNs) with pre-trained weights and conversion scripts. | 206 |