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.☆274Updated last year
- PyTorch Dataset Rank Dataset☆43Updated 4 years ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆84Updated 5 years ago
- OneFlow models for benchmarking.☆104Updated last year
- Simple Dynamic Batching Inference☆145Updated 3 years ago
- ☆18Updated 2 years ago
- Simple tutorials on Pytorch DDP training☆281Updated 3 years ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆408Updated last month
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆200Updated 2 years ago
- Models and examples built with OneFlow☆98Updated 10 months ago
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 3 years ago
- ☆220Updated 2 years ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆92Updated 2 years ago
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆232Updated 4 years ago
- A PyTorch toolkit for extremely fast ImageNet training with NVIDIA DALI.☆51Updated 4 years ago
- ☆196Updated last year
- The record of what I‘ve been through.☆101Updated 7 months ago
- ☆37Updated 2 years ago
- TensorRT 2022复赛方案: 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆141Updated 3 years ago
- mini dataloader☆15Updated 3 years ago
- ☆36Updated 3 years ago
- DeepLearning Framework Performance Profiling Toolkit☆288Updated 3 years ago
- Group Fisher Pruning for Practical Network Compression(ICML2021)☆160Updated 2 years ago
- ☆99Updated 4 years ago
- A brief of TorchScript by MNIST☆112Updated 3 years ago
- Code release for "Self-Tuning for Data-Efficient Deep Learning" (ICML 2021)☆112Updated 4 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆69Updated 2 years ago
- (CVPR 2021, Oral) Dynamic Slimmable Network☆229Updated 3 years ago
- Implementation of Neuron-level Structured Pruning using Polarization Regularizer☆83Updated 2 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆132Updated 2 years ago