Pinging-ZJU / DNN-PrinterLinks
Python Scritpt which can be embedded into PyTorch model to print the model size.
☆19Updated 4 years ago
Alternatives and similar repositories for DNN-Printer
Users that are interested in DNN-Printer are comparing it to the libraries listed below
Sorting:
- The pure and clear PyTorch Distributed Training Framework.☆275Updated last year
- Simple tutorials on Pytorch DDP training☆285Updated 3 years ago
- PyTorch Dataset Rank Dataset☆43Updated 4 years ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆406Updated 5 months ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆84Updated 5 years ago
- ☆18Updated 3 years ago
- ☆36Updated 2 years ago
- The record of what I‘ve been through. Now moved to Notion. See link below☆102Updated 11 months ago
- OneFlow models for benchmarking.☆104Updated last year
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆232Updated 5 years ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆92Updated 3 years ago
- Models and examples built with OneFlow☆101Updated last year
- ☆196Updated last year
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆199Updated 3 years ago
- ☆59Updated 3 years ago
- WuDaoMM this is a data project☆74Updated 3 years ago
- ☆168Updated 2 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆70Updated 3 years ago
- Simple Dynamic Batching Inference☆145Updated 3 years ago
- DeepLearning Framework Performance Profiling Toolkit☆294Updated 3 years ago
- Product1M☆90Updated 3 years ago
- EssentialMC2 Video Understanding.☆115Updated 3 years ago
- tinynn with automatic differentiation☆40Updated 2 years ago
- ☆219Updated 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…☆185Updated 2 years ago
- Code release for "Self-Tuning for Data-Efficient Deep Learning" (ICML 2021)☆112Updated 4 years ago
- 空闲GPU和CPU占用程序☆38Updated last year
- TaiSu(太素)--a large-scale Chinese multimodal dataset(亿级大规模中文视觉语言预训练数据集)☆189Updated 2 years ago
- A PyTorch toolkit for extremely fast ImageNet training with NVIDIA DALI.☆51Updated 5 years ago
- Yet another PyTorch Trainer and some core components for deep learning.☆222Updated last year