zhijian-liu / torchprofile
A general and accurate MACs / FLOPs profiler for PyTorch models
☆569Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for torchprofile
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.☆423Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆296Updated 3 years ago
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆201Updated last year
- PyTorch layer-by-layer model profiler☆608Updated 3 years ago
- ☆223Updated 3 years ago
- [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework☆274Updated 11 months ago
- ☆214Updated 2 years ago
- Low Precision Arithmetic Simulation in PyTorch☆264Updated 5 months ago
- Collection of recent methods on (deep) neural network compression and acceleration.☆927Updated 2 months ago
- [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing☆328Updated 3 months ago
- Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019☆910Updated last year
- Pruning Neural Networks with Taylor criterion in Pytorch☆312Updated 5 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆580Updated last year
- A GPU performance profiling tool for PyTorch models☆493Updated 3 years ago
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆428Updated 11 months ago
- A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…☆606Updated last year
- Demystify RAM Usage in Multi-Process Data Loaders☆178Updated last year
- My best practice of training large dataset using PyTorch.☆1,087Updated 6 months ago
- A LARS implementation in PyTorch☆335Updated 4 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- PC-DARTS:Partial Channel Connections for Memory-Efficient Differentiable Architecture Search☆436Updated 4 years ago
- Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.☆413Updated last year
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆693Updated 2 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆165Updated 2 years ago
- Papers for deep neural network compression and acceleration☆394Updated 3 years ago
- Pytorch implementation of BRECQ, ICLR 2021☆250Updated 3 years ago
- Mobile vision models and code☆901Updated 4 months ago
- PyTorch Implementation of DARTS: Differentiable Architecture Search☆439Updated 2 years ago
- Summary, Code for Deep Neural Network Quantization☆530Updated 3 weeks ago