zhijian-liu / torchprofileView external linksLinks
A general and accurate MACs / FLOPs profiler for PyTorch models
☆635Jul 29, 2025Updated 6 months ago
Alternatives and similar repositories for torchprofile
Users that are interested in torchprofile are comparing it to the libraries listed below
Sorting:
- [CVPR 2019, Oral] HAQ: Hardware-Aware Automated Quantization with Mixed Precision☆402Feb 26, 2021Updated 4 years ago
- [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment☆1,939Dec 14, 2023Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing☆337Jul 14, 2024Updated last year
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆449Nov 22, 2023Updated 2 years ago
- [ICLR 2019] ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardware☆1,449Aug 30, 2024Updated last year
- PyTorch layer-by-layer model profiler☆607May 23, 2021Updated 4 years ago
- Mobile vision models and code☆916Updated this week
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆928Mar 9, 2023Updated 2 years ago
- Single-Path NAS: Designing Hardware-Efficient ConvNets in less than 4 Hours☆396Dec 14, 2020Updated 5 years ago
- Flops counter for neural networks in pytorch framework☆2,970Aug 20, 2025Updated 5 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,224Jan 12, 2026Updated last month
- (CVPR 2021, Oral) Dynamic Slimmable Network☆231Dec 31, 2021Updated 4 years ago
- Automated deep learning algorithms implemented in PyTorch.☆1,584Apr 24, 2022Updated 3 years ago
- ☆265Oct 30, 2019Updated 6 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Jul 4, 2020Updated 5 years ago
- Densely Connected Search Space for More Flexible Neural Architecture Search (CVPR2020)☆294Jul 4, 2020Updated 5 years ago
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆168Feb 26, 2021Updated 4 years ago
- Automated Deep Learning: Neural Architecture Search Is Not the End (a curated list of AutoDL resources and an in-depth analysis)☆2,336Sep 26, 2022Updated 3 years ago
- Profiling and inspecting memory in pytorch☆1,078Sep 5, 2025Updated 5 months ago
- A neural network training interface based on PyTorch, with a focus on flexibility☆63Jan 17, 2024Updated 2 years ago
- NASBench: A Neural Architecture Search Dataset and Benchmark☆710May 1, 2023Updated 2 years ago
- [CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs☆1,118Jun 5, 2024Updated last year
- ☆1,510Aug 27, 2020Updated 5 years ago
- Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)☆1,516Jun 7, 2020Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆509Jul 13, 2021Updated 4 years ago
- Differentiable architecture search for convolutional and recurrent networks☆3,993Jan 3, 2021Updated 5 years ago
- [IROS 2021] ADD: A Fine-grained Dynamic Inference Architecture for Semantic Image Segmentation☆10May 3, 2022Updated 3 years ago
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,169Mar 20, 2024Updated last year
- A curated list of neural network pruning resources.☆2,490Apr 4, 2024Updated last year
- MetaPruning: Meta Learning for Automatic Neural Network Channel Pruning. In ICCV 2019.☆352Jul 5, 2020Updated 5 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆602Dec 27, 2019Updated 6 years ago
- ☆150Jul 18, 2020Updated 5 years ago
- Codes for our paper "Progressive Differentiable Architecture Search:Bridging the Depth Gap between Search and Evaluation"☆364Feb 18, 2020Updated 5 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,614Jun 16, 2021Updated 4 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆305May 20, 2022Updated 3 years ago
- [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy☆160Jun 16, 2020Updated 5 years ago
- (ECCV'2020 Oral)EagleEye: Fast Sub-net Evaluation for Efficient Neural Network Pruning☆309Dec 8, 2022Updated 3 years ago