bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
☆84Updated 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
- Synchronized Multi-GPU Batch Normalization☆67Updated 6 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆68Updated 2 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆77Updated 6 years ago
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Updated 5 years ago
- ☆68Updated 5 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆102Updated 5 years ago
- A package to make do Network Slimming a little easier☆47Updated 3 years ago
- ☆120Updated 4 years ago
- A Simple & Flexible Cross Framework Operators Toolkit☆164Updated 4 years ago
- ☆166Updated 2 years ago
- CUDA implementation of NMS for PyTorch☆86Updated 5 years ago
- Personal Pytorch toy script.☆67Updated 3 years ago
- Accelerated Training for Massive Classification via Dynamic Class Selection (AAAI 2018, Oral)☆99Updated 4 years ago
- ☆45Updated 5 years ago
- Caffe implementation of ICCV 2017 & TPAMI 2018 paper - ThiNet☆46Updated 6 years ago
- A tool to count operators and parameters of your MXNet-Gluon model.☆23Updated 5 years ago
- KnowledgeDistillation Layer (Caffe implementation)☆89Updated 7 years ago
- Android demo for dabnn☆20Updated 5 years ago
- Caffe implementation of Optimal-Ternary-Weights-Approximation in "Two-Step Quantization for Low-bit Neural Networks" (CVPR2018).☆14Updated 6 years ago
- Implement Towards Effective Low-bitwidth Convolutional Neural Networks☆41Updated 6 years ago
- Code for IJCAI2019 paper☆46Updated 5 years ago
- ☆99Updated 3 years ago
- Decode JPEG image on GPU using PyTorch☆90Updated last year
- Simulate quantization and quantization aware training for MXNet-Gluon models.☆46Updated 5 years ago
- alibabacloud-quantization-networks☆121Updated 5 years ago
- Rethinking the Smaller-Norm-Less-Informative Assumption in Channel Pruning of Convolution Layers https://arxiv.org/abs/1802.00124☆72Updated 6 years ago
- ☆135Updated 6 years ago
- Model shared. Top1:67.408/Top5:87.258☆48Updated 6 years ago
- [ICLR 2020]: 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'☆222Updated 4 years ago
- network slimming with pytorch☆24Updated 4 years ago