THU-MIG / torch-model-compressionLinks
针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库
☆250Updated 2 years ago
Alternatives and similar repositories for torch-model-compression
Users that are interested in torch-model-compression are comparing it to the libraries listed below
Sorting:
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆212Updated last year
- ONNX2Pytorch☆162Updated 4 years ago
- pytorch AutoSlim tools，支持三行代码对pytorch模型进行剪枝压缩☆39Updated 4 years ago
- ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgetting (ICCV 2021)☆296Updated 2 years ago
- model compression based on pytorch (1、quantization: 8/4/2bits(dorefa)、ternary/binary value(twn/bnn/xnor-net)；2、 pruning: normal、regular a…☆170Updated 5 years ago
- Offline Quantization Tools for Deploy.☆131Updated 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 2 years ago
- www.giantpandacv.com☆151Updated last year
- ☆99Updated 4 years ago
- Model Compression 1. Pruning(BN Pruning) 2. Knowledge Distillation (Hinton) 3. Quantization (MNN) 4. Deployment (MNN)☆80Updated 4 years ago
- Everything in Torch Fx☆344Updated last year
- A simple network quantization demo using pytorch from scratch.☆536Updated 2 years ago
- tensorrt int8 量化yolov5 onnx模型☆185Updated 4 years ago
- An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.☆225Updated 3 years ago
- 模型压缩demo（剪枝、量化、知识蒸馏）☆77Updated 5 years ago
- Official repo of RepOptimizers and RepOpt-VGG☆264Updated 2 years ago
- EasyQuant(EQ) is an efficient and simple post-training quantization method via effectively optimizing the scales of weights and activatio…☆402Updated 2 years ago
- An Improved One millisecond Mobile Backbone☆145Updated 2 years ago
- A more efficient yolov5 with oneflow backend 🎉🎉🎉☆215Updated 3 weeks ago
- TensorRT 2022复赛方案： 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆140Updated 3 years ago
- ☆146Updated last year
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆198Updated 3 years ago
- RepVGG TensorRT int8 量化，实测推理不到1ms一帧！☆63Updated 4 years ago
- Model Quantization Benchmark☆826Updated 3 months ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队：美迪康 AI Lab☆171Updated 3 years ago
- Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.☆299Updated last week
- A nnie quantization aware training tool on pytorch.☆239Updated 4 years ago
- NVIDIA TensorRT 加速推断教程！☆133Updated 4 years ago
- This is 8-bit quantization sample for yolov5. Both PTQ, QAT and Partial Quantization have been implemented, and present the results based…☆105Updated 3 years ago
- Trans different platform's network to International Representation(IR)☆44Updated 7 years ago