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