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
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 |