THU-MIG / torch-model-compression
针对pytorch模型的自动化模型结构分析和修改工具集,包含自动分析模型结构的模型压缩算法库
☆242Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torch-model-compression
- model compression based on pytorch (1、quantization: 8/4/2bits(dorefa)、ternary/binary value(twn/bnn/xnor-net);2、 pruning: normal、regular a…☆170Updated 4 years ago
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆191Updated 11 months ago
- Model Compression 1. Pruning(BN Pruning) 2. Knowledge Distillation (Hinton) 3. Quantization (MNN) 4. Deployment (MNN)☆77Updated 3 years ago
- Model Quantization Benchmark☆762Updated 5 months ago
- A simple network quantization demo using pytorch from scratch.☆506Updated last year
- RM Operation can equivalently convert ResNet to VGG, which is better for pruning; and can help RepVGG perform better when the depth is la…☆211Updated last year
- pytorch AutoSlim tools,支持三行代码对pytorch模型进行剪枝压缩☆37Updated 3 years ago
- ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgetting (ICCV 2021)☆288Updated last year
- ONNX2Pytorch☆158Updated 3 years ago
- An Improved One millisecond Mobile Backbone☆142Updated 2 years ago
- An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.☆217Updated 2 years ago
- tensorrt int8 量化yolov5 onnx模型☆176Updated 3 years ago
- Official repo of RepOptimizers and RepOpt-VGG☆252Updated last year
- www.giantpandacv.com☆149Updated 4 months ago
- Offline Quantization Tools for Deploy.☆116Updated 10 months ago
- Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.☆288Updated 8 months ago
- B站Efficient-Neural-Network学习分享的配套代码☆296Updated 2 years ago
- OpenPose uses Pytorch for static quantization, saving, and loading of models☆78Updated 3 years ago
- A nnie quantization aware training tool on pytorch.☆238Updated 3 years ago
- Everything in Torch Fx☆341Updated 5 months ago
- EasyQuant(EQ) is an efficient and simple post-training quantization method via effectively optimizing the scales of weights and activatio…☆392Updated last year
- ☆96Updated 3 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆191Updated 2 years ago
- PyTorch Neural Network eXchange☆517Updated 2 weeks ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆163Updated 2 years ago
- Some tools to operate PaddlePaddle model☆72Updated 2 years ago
- This repository is Onnx tutorial summary for python implements , which comes from other web resource.☆25Updated 2 years ago
- TensorRT 2022复赛方案: 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆135Updated 2 years ago
- NVIDIA TensorRT 加速推断教程!☆132Updated 3 years ago