VainF / Torch-Pruning
[CVPR 2023] DepGraph: Towards Any Structural Pruning
☆2,836Updated 3 weeks ago
Alternatives and similar repositories for Torch-Pruning:
Users that are interested in Torch-Pruning are comparing it to the libraries listed below
- A curated list of neural network pruning resources.☆2,389Updated 9 months ago
- OpenMMLab Model Compression Toolbox and Benchmark.☆1,510Updated 7 months ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,360Updated last year
- A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆25 knowledge distillation methods p…☆1,423Updated this week
- This is a collection of our NAS and Vision Transformer work.☆1,707Updated 5 months ago
- Flops counter for convolutional networks in pytorch framework☆2,848Updated 3 months ago
- Efficient computing methods developed by Huawei Noah's Ark Lab☆1,228Updated 2 months ago
- EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]☆1,006Updated last year
- Count the MACs / FLOPs of your PyTorch model.☆4,940Updated 6 months ago
- A list of papers, docs, codes about model quantization. This repo is aimed to provide the info for model quantization research, we are co…☆1,958Updated 2 months ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,115Updated last month
- Pytorch implementation of various Knowledge Distillation (KD) methods.☆1,644Updated 3 years ago
- The implementation of various lightweight networks by using PyTorch. such as：MobileNetV2，MobileNeXt，GhostNet，ParNet，MobileViT、AdderNet，Sh…☆841Updated 2 years ago
- Deformable DETR: Deformable Transformers for End-to-End Object Detection.☆3,360Updated 8 months ago
- micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantiz…☆2,227Updated 3 years ago
- Code release for ConvNeXt V2 model☆1,584Updated 5 months ago
- The official implementation of [CVPR2022] Decoupled Knowledge Distillation https://arxiv.org/abs/2203.08679 and [ICCV2023] DOT: A Distill…☆825Updated last year
- [ICLR 2020] Contrastive Representation Distillation (CRD), and benchmark of recent knowledge distillation methods☆2,239Updated last year
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,364Updated last year
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,610Updated 9 months ago
- detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.☆2,072Updated 5 months ago
- CVNets: A library for training computer vision networks☆1,818Updated last year
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,057Updated last month
- A PyTorch implementation for exploring deep and shallow knowledge distillation (KD) experiments with flexibility☆1,887Updated last year
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,647Updated this week
- Codebase for Image Classification Research, written in PyTorch.☆2,143Updated 9 months ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and …☆1,830Updated last year
- RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything☆872Updated 7 months ago
- Simplify your onnx model☆3,942Updated 4 months ago
- [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment☆1,893Updated last year