zugexiaodui / torch_flops
A library for calculating the FLOPs in the forward() process based on torch.fx
☆72Updated 2 weeks ago
Related projects: ⓘ
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆238Updated last year
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆283Updated last month
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆64Updated 6 months ago
- [CVPR'23] SparseViT: Revisiting Activation Sparsity for Efficient High-Resolution Vision Transformer☆59Updated 4 months ago
- Official implementation of paper "Knowledge Distillation from A Stronger Teacher", NeurIPS 2022☆133Updated last year
- [NeurIPS 2023] Structural Pruning for Diffusion Models☆156Updated 2 months ago
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Updated last year
- [ICCV23] Robust Mixture-of-Expert Training for Convolutional Neural Networks by Yihua Zhang, Ruisi Cai, Tianlong Chen, Guanhua Zhang, Hua…☆39Updated last year
- Recent Advances on Efficient Vision Transformers☆46Updated last year
- PyTorch code and checkpoints release for VanillaKD: https://arxiv.org/abs/2305.15781☆71Updated 10 months ago
- The official implementation of the NeurIPS 2022 paper Q-ViT.☆77Updated last year
- Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)☆121Updated last year
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "Fast Vision Transformers with HiLo Attention"☆239Updated 10 months ago
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆63Updated 3 weeks ago
- Awasome Papers and Resources in Deep Neural Network Pruning with Source Code.☆119Updated 3 weeks ago
- Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.☆77Updated 6 months ago
- [ECCV 2022] EdgeViT: Competing Light-weight CNNs on Mobile Devices with Vision Transformers☆97Updated last year
- Official code for paper "On the Connection between Local Attention and Dynamic Depth-wise Convolution" ICLR 2022 Spotlight☆182Updated last year
- Official Code of Paper "Reversible Column Networks" "RevColv2"☆246Updated last year
- [ICCV 23]An approach to enhance the efficiency of Vision Transformer (ViT) by concurrently employing token pruning and token merging tech…☆85Updated last year
- [CVPR 2024] PTQ4SAM: Post-Training Quantization for Segment Anything☆50Updated 2 months ago
- PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)☆75Updated last year
- Code Implementation of EfficientVMamba☆172Updated 5 months ago
- ☆170Updated 9 months ago
- PyTorch code and checkpoints release for OFA-KD: https://arxiv.org/abs/2310.19444☆89Updated 5 months ago
- Unofficial Implementation of MLP-Mixer, gMLP, resMLP, Vision Permutator, S2MLP, S2MLPv2, RaftMLP, HireMLP, ConvMLP, AS-MLP, SparseMLP, Co…☆161Updated 2 years ago
- Official repository of Slide-Transformer (CVPR2023)☆157Updated 3 weeks ago
- Official implement for ICML2023 paper: "A Closer Look at Self-Supervised Lightweight Vision Transformers"☆109Updated last year
- [IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer☆301Updated last year
- BinaryViT: Pushing Binary Vision Transformers Towards Convolutional Models☆26Updated 7 months ago