packnet

Task Network Pruning Framework

A deep learning framework for adding multiple tasks to a single network by iterative pruning and evaluation on various datasets.

Code for PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruning

GitHub

230 stars
7 watching
40 forks
Language: Python
last commit: about 6 years ago

Related projects:

Repository Description Stars
arunmallya/piggyback Adapting a single network to multiple tasks by learning to mask weights 182
ntumslab/prune A generative approach for network embedding that preserves proximity and global ranking. 45
alecwangcq/eigendamage-pytorch A deep learning project implementing structured pruning algorithms in PyTorch for efficient neural network training and inference. 112
neuralmagic/sparseml Enables the creation of smaller neural network models through efficient pruning and quantization techniques 2,071
prinsphield/adversarial_reprogramming This project enables reprogramming of pre-trained neural networks to work on new tasks by fine-tuning them on smaller datasets. 33
rocm/mxnet A deep learning framework that enables efficient and flexible distributed/mobile deep learning with dynamic dataflow dependency scheduling 28
taolei87/rcnn An implementation of neural network components and optimization methods for text analysis, including rationales for neural predictions. 355
hyeonwoonoh/decouplednet A deep learning framework for semantic segmentation using pre-trained classification networks and heterogeneous annotations 74
tusimple/sparse-structure-selection Re-implements sparse structure selection algorithm for deep neural networks in a modified MXNet framework. 87
jacobgil/pytorch-pruning This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. 875
microsoft/archai Automates the search for optimal neural network configurations in deep learning applications 467
mrkn/mxnet.rb A Ruby interface to MXNet's deep learning framework 48
taehoonlee/tensornets A collection of pre-trained neural network models with simple interfaces for easy integration into machine learning workflows. 1,004
fedml-ai/spreadgnn A framework for decentralized multi-task learning of graph neural networks on molecular data with guaranteed convergence 44
bayeswatch/pytorch-prunes A tool for training neural networks with pruned weights and evaluating their performance. 139