bindog / pytorch-model-parallel
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
☆79Updated 4 years ago
Related projects: ⓘ
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Updated 4 years ago
- Personal Pytorch toy script.☆66Updated 2 years ago
- ☆119Updated 4 years ago
- [ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions, surpassing MobileNetV2☆103Updated 4 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆66Updated last year
- ☆107Updated this week
- CUDA implementation of NMS for PyTorch☆86Updated 4 years ago
- ☆68Updated 5 years ago
- ☆45Updated 5 years ago
- A package to make do Network Slimming a little easier☆47Updated 2 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆76Updated 5 years ago
- Face Recognition Loss on Mnist, Pytorch implementation☆93Updated 3 years ago
- Synchronized Multi-GPU Batch Normalization☆67Updated 6 years ago
- Accelerated Training for Massive Classification via Dynamic Class Selection (AAAI 2018)☆96Updated 4 years ago
- pytorch CircleLoss for Face recognition☆26Updated 4 years ago
- [ICLR 2020]: 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'☆224Updated 4 years ago
- ☆128Updated this week
- network slimming with pytorch☆24Updated 4 years ago
- Caffe implementation of ICCV 2017 & TPAMI 2018 paper - ThiNet☆46Updated 6 years ago
- Decode JPEG image on GPU using PyTorch☆83Updated 11 months ago
- ☆134Updated 5 years ago
- Simulate quantization and quantization aware training for MXNet-Gluon models.☆46Updated 4 years ago
- Rethinking the Smaller-Norm-Less-Informative Assumption in Channel Pruning of Convolution Layers https://arxiv.org/abs/1802.00124☆71Updated 5 years ago
- Apply the pruning strategy for MobileNet_v2☆51Updated 5 years ago
- A Simple & Flexible Cross Framework Operators Toolkit☆164Updated 3 years ago
- Codes of Centripetal SGD☆62Updated 2 years ago
- PyTorch reimplementation of RegNet (Design Space Design, CVPR2020) on CIFAR10 and ImageNet☆46Updated 4 years ago
- Code and model for the winner team 'trojans' in ICCV19-Lightweight Face Recognition Challenge☆127Updated 4 years ago
- ☆165Updated last year
- A tool to count operators and parameters of your MXNet-Gluon model.☆21Updated 4 years ago