bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
☆82Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for pytorch-model-parallel
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆66Updated last year
- ☆120Updated 4 years ago
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Updated 4 years ago
- ☆68Updated 5 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆77Updated 5 years ago
- A package to make do Network Slimming a little easier☆47Updated 2 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆102Updated 4 years ago
- Personal Pytorch toy script.☆67Updated 2 years ago
- ☆96Updated 3 years ago
- https://arxiv.org/abs/2005.10497☆73Updated 4 years ago
- Synchronized Multi-GPU Batch Normalization☆67Updated 6 years ago
- Decode JPEG image on GPU using PyTorch☆84Updated last year
- Face Recognition Loss on Mnist, Pytorch implementation☆94Updated 3 years ago
- Apply the pruning strategy for MobileNet_v2☆51Updated 5 years ago
- ☆45Updated 5 years ago
- A nnie quantization aware training tool on pytorch.☆238Updated 3 years ago
- PyTorch Dataset Rank Dataset☆40Updated 3 years ago
- This is Top 3 Code for the Person ReID Compitition of NAIC☆150Updated 4 years ago
- alibabacloud-quantization-networks☆121Updated 5 years ago
- Base to channel pruned to ResNet18 model☆143Updated 2 years ago
- Accelerated Training for Massive Classification via Dynamic Class Selection (AAAI 2018)☆96Updated 4 years ago
- A tool to count operators and parameters of your MXNet-Gluon model.☆21Updated 4 years ago
- Implement Towards Effective Low-bitwidth Convolutional Neural Networks☆42Updated 6 years ago
- Code and model for the winner team 'trojans' in ICCV19-Lightweight Face Recognition Challenge☆127Updated 5 years ago
- Rethinking the Smaller-Norm-Less-Informative Assumption in Channel Pruning of Convolution Layers https://arxiv.org/abs/1802.00124☆71Updated 6 years ago
- quantize aware training package for NCNN on pytorch☆68Updated 3 years ago
- [ICLR 2020]: 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'☆223Updated 4 years ago
- Codes of Centripetal SGD☆62Updated 2 years ago