zugexiaodui / torch_flops
A library for calculating the FLOPs in the forward() process based on torch.fx
☆81Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for torch_flops
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆329Updated 3 months ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆71Updated 8 months ago
- [NeurIPS 2023] Structural Pruning for Diffusion Models☆164Updated 4 months ago
- Implementation of Post-training Quantization on Diffusion Models (CVPR 2023)☆125Updated last year
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆244Updated last year
- The official implementation of the NeurIPS 2022 paper Q-ViT.☆83Updated last year
- Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures☆371Updated 3 weeks ago
- [IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer☆308Updated last year
- QuEST: Efficient Finetuning for Low-bit Diffusion Models☆34Updated 3 months ago
- 📚 Collection of awesome generation acceleration resources.☆43Updated last week
- ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation☆32Updated 2 months ago
- ☆181Updated 11 months ago
- ☆100Updated last month
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆213Updated 3 weeks ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆50Updated 5 months ago
- Awesome list of papers that extend Mamba to various applications.☆128Updated 2 months ago
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆366Updated this week
- Awasome Papers and Resources in Deep Neural Network Pruning with Source Code.☆134Updated 2 months ago
- Code Implementation of EfficientVMamba☆183Updated 7 months ago
- Post-Training Quantization for Vision transformers.☆190Updated 2 years ago
- [ECCV 2024] Isomorphic Pruning for Vision Models☆53Updated 3 months ago
- [ICCV 23]An approach to enhance the efficiency of Vision Transformer (ViT) by concurrently employing token pruning and token merging tech…☆89Updated last year
- [CVPR 2023] PD-Quant: Post-Training Quantization Based on Prediction Difference Metric☆51Updated last year
- The official implementation of the ICML 2023 paper OFQ-ViT☆27Updated last year
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆79Updated 2 months ago
- A curated list of papers on the applications of RWKV in computer vision.☆112Updated 2 weeks ago
- 1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundatio…☆209Updated 2 months ago
- [CVPR 2024] PTQ4SAM: Post-Training Quantization for Segment Anything☆56Updated 4 months ago
- Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.☆81Updated 8 months ago
- The official implementation of PTQD: Accurate Post-Training Quantization for Diffusion Models☆88Updated 8 months ago