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 2 years ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆84Updated 5 years ago
- PyTorch Dataset Rank Dataset☆43Updated 4 years ago
- OneFlow models for benchmarking.☆104Updated last year
- Simple tutorials on Pytorch DDP training☆286Updated 3 years ago
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆198Updated 3 years ago
- Models and examples built with OneFlow☆101Updated last year
- ☆18Updated 3 years ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆405Updated 6 months ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆92Updated 3 years ago
- Simple Dynamic Batching Inference☆145Updated 3 years ago
- Datasets, Transforms and Models specific to Computer Vision☆90Updated 2 years ago
- ☆12Updated 2 years ago
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 4 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Updated 2 years ago
- ☆196Updated last year
- ☆13Updated 2 years ago
- Code for "Searching for Efficient Multi-Stage Vision Transformers"☆63Updated 4 years ago
- The record of what I‘ve been through. Now moved to Notion. See link below☆102Updated last year
- Implementation of Neuron-level Structured Pruning using Polarization Regularizer☆83Updated 3 years ago
- (CVPR 2021, Oral) Dynamic Slimmable Network☆231Updated 4 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
- Group Fisher Pruning for Practical Network Compression(ICML2021)☆161Updated 2 years ago
- Trans different platform's network to International Representation(IR)☆44Updated 7 years ago
- Code for "DARTS-: Robustly Stepping out of Performance Collapse Without Indicators"☆57Updated 5 years ago
- mini dataloader☆15Updated 3 years ago
- https://hyperbox-doc.readthedocs.io/en/latest/☆26Updated 2 years ago
- ☆37Updated 3 years ago
- ☆219Updated 2 years ago
- WuDaoMM this is a data project☆74Updated 3 years ago