Pinging-ZJU / DNN-Printer
Python Scritpt which can be embedded into PyTorch model to print the model size.
☆18Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for DNN-Printer
- Models and examples built with OneFlow☆96Updated 3 weeks ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆82Updated 4 years ago
- PyTorch Dataset Rank Dataset☆40Updated 3 years ago
- ☆28Updated last year
- ☆34Updated 2 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆66Updated last year
- Personal Pytorch toy script.☆67Updated 2 years ago
- A general framework for inferring CNNs efficiently. Reduce the inference latency of MobileNet-V3 by 1.3x on an iPhone XS Max without sac…☆181Updated last year
- Code for "Searching for Efficient Multi-Stage Vision Transformers"☆62Updated 3 years ago
- ☆74Updated 10 months ago
- OneFlow models for benchmarking.☆104Updated 3 months ago
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆201Updated last year
- ☆11Updated last year
- Simple Dynamic Batching Inference☆145Updated 2 years ago
- CVPR 2021 : Zero-shot Adversarial Quantization (ZAQ)☆65Updated 3 years ago
- Trans different platform's network to International Representation(IR)☆44Updated 6 years ago
- The pure and clear PyTorch Distributed Training Framework.☆276Updated 9 months ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆89Updated last year
- The record of what I‘ve been through.☆95Updated 3 weeks ago
- tinynn with automatic differentiation☆38Updated last year
- A package to make do Network Slimming a little easier☆47Updated 2 years ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆57Updated last year
- ☆96Updated 3 years ago
- Simple tutorials on Pytorch DDP training☆262Updated 2 years ago
- Gpu 任务排队☆2Updated last year
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆129Updated last year
- Möbius Transformation for Fast Inner Product Search on Graph☆23Updated 3 years ago
- Datasets, Transforms and Models specific to Computer Vision☆82Updated 11 months ago
- General Image Classification Code base☆21Updated 3 years ago