bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
☆85Updated 4 years ago
Alternatives and similar repositories for pytorch-model-parallel:
Users that are interested in pytorch-model-parallel are comparing it to the libraries listed below
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Updated 4 years ago
- Synchronized Multi-GPU Batch Normalization☆67Updated 6 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆102Updated 4 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆67Updated 2 years ago
- Personal Pytorch toy script.☆67Updated 2 years ago
- ☆120Updated 4 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆77Updated 6 years ago
- A Simple & Flexible Cross Framework Operators Toolkit☆164Updated 4 years ago
- ☆97Updated 3 years ago
- A package to make do Network Slimming a little easier☆47Updated 3 years ago
- [ICLR 2020]: 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'☆222Updated 4 years ago
- ☆45Updated 5 years ago
- ☆68Updated 5 years ago
- CUDA implementation of NMS for PyTorch☆85Updated 5 years ago
- A nnie quantization aware training tool on pytorch.☆238Updated 4 years ago
- MoGA: Searching Beyond MobileNetV3☆228Updated 4 years ago
- ☆23Updated 9 months ago
- ☆165Updated last year
- Simulate quantization and quantization aware training for MXNet-Gluon models.☆46Updated 4 years ago
- ☆12Updated 2 years ago
- ☆262Updated 5 years ago
- Face Recognition Loss on Mnist, Pytorch implementation☆94Updated 4 years ago
- A tool to count operators and parameters of your MXNet-Gluon model.☆22Updated 4 years ago
- pytorch CircleLoss for Face recognition☆25Updated 4 years ago
- Implement Towards Effective Low-bitwidth Convolutional Neural Networks☆42Updated 6 years ago
- https://arxiv.org/abs/2005.10497☆73Updated 4 years ago
- This is Top 3 Code for the Person ReID Compitition of NAIC☆150Updated 4 years ago
- OICSR: Out-In-Channel Sparsity Regularization for Compact Deep Neural Networks☆26Updated 5 years ago
- Apply the pruning strategy for MobileNet_v2☆51Updated 5 years ago
- Model shared. Top1:67.408/Top5:87.258☆48Updated 6 years ago