BBuf / model-compression
model compression based on pytorch (1、quantization: 8/4/2bits(dorefa)、ternary/binary value(twn/bnn/xnor-net);2、 pruning: normal、regular and group convolutional channel pruning;3、 group convolution structure;4、batch-normalization folding for binary value of feature(A))
☆170Updated 4 years ago
Alternatives and similar repositories for model-compression:
Users that are interested in model-compression are comparing it to the libraries listed below
- 针对pytorch模型的自动化模型结构分析和修改工具集,包含自动分析模型结构的模型压缩算法库☆248Updated last year
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆206Updated last year
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆243Updated 4 years ago
- pytorch AutoSlim tools,支持三行代码对pytorch模型进行剪枝压缩☆38Updated 3 years ago
- www.giantpandacv.com☆147Updated 8 months ago
- A nnie quantization aware training tool on pytorch.☆239Updated 4 years ago
- ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgetting (ICCV 2021)☆294Updated 2 years ago
- Personal Pytorch toy script.☆67Updated 2 years ago
- RM Operation can equivalently convert ResNet to VGG, which is better for pruning; and can help RepVGG perform better when the depth is la…☆212Updated last year
- ONNX2Pytorch☆160Updated 3 years ago
- Try out different pruning-approaches on lightweight Backbones.☆146Updated 4 years ago
- An Improved One millisecond Mobile Backbone☆144Updated 2 years ago
- A simple network quantization demo using pytorch from scratch.☆522Updated last year
- Everything in Torch Fx☆342Updated 9 months ago
- ghostnet_cifar10☆114Updated 4 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆160Updated 4 years ago
- An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.☆222Updated 3 years ago
- OpenPose uses Pytorch for static quantization, saving, and loading of models☆85Updated 3 years ago
- ☆32Updated 4 years ago
- RepVGG TensorRT int8 量化,实测推理不到1ms一帧!☆62Updated 3 years ago
- Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.☆568Updated 5 years ago
- tensorrt int8 量化yolov5 onnx模型☆181Updated 3 years ago
- 模型压缩demo(剪枝、量化、知识蒸馏)☆71Updated 4 years ago
- ☆79Updated 4 years ago
- ☆97Updated 3 years ago
- ☆62Updated 4 years ago
- NVIDIA TensorRT 加速推断教程!☆133Updated 3 years ago
- yolov5 onnx caffe☆115Updated 3 years ago
- Network reproduction and expansion based on Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017).☆24Updated 2 years ago
- Model Quantization Benchmark☆790Updated last month