A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
☆84Jun 11, 2020Updated 5 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
Sorting:
- Code for the ICML 2021 paper "Sharing Less is More: Lifelong Learning in Deep Networks with Selective Layer Transfer"☆12Aug 17, 2021Updated 4 years ago
- Python bindings for minimp3☆17Sep 11, 2023Updated 2 years ago
- Rotate RoI Align and Rotate Position Sensitive RoI Align Operation in Caffe☆15Dec 5, 2018Updated 7 years ago
- ☆20Dec 10, 2018Updated 7 years ago
- Caffe re-implementation of dynamic network surgery.☆18Jun 15, 2018Updated 7 years ago
- ☆187Jan 21, 2021Updated 5 years ago
- ☆10Oct 23, 2017Updated 8 years ago
- A simple baseline for Person ReID, it achieves 3rd place in VisDA2020 challenge.☆38Aug 21, 2020Updated 5 years ago
- [IROS 2021] ADD: A Fine-grained Dynamic Inference Architecture for Semantic Image Segmentation☆10May 3, 2022Updated 3 years ago
- The demo for "Discretization and Re-synthesis: an alternative method to solve the Cocktail Party Problem".☆12Oct 25, 2021Updated 4 years ago
- Face Detection Based on Detectron2☆59Jun 16, 2020Updated 5 years ago
- Pytorch based implemenatation of Fully Convolutional Networks☆19Aug 11, 2018Updated 7 years ago
- Implementation of popular deep learning networks with TensorRT network definition APIs☆10Mar 25, 2021Updated 4 years ago
- Learning to Cluster Faces (CVPR 2019, CVPR 2020)☆722Dec 27, 2021Updated 4 years ago
- Code for CVPR2019 paper《Unequal Training for Deep Face Recognition with Long Tailed Noisy Data》☆69Jun 24, 2019Updated 6 years ago
- Improving Face Recognition from Hard Samplesvia Distribution Distillation Loss☆46May 24, 2021Updated 4 years ago
- ☆35Nov 19, 2018Updated 7 years ago
- The quantization of CNN/LSTM☆11Mar 26, 2017Updated 8 years ago
- TensorRT half precision inference routine on a API-based TensorRT model☆13Jul 3, 2018Updated 7 years ago
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Mar 27, 2020Updated 5 years ago
- Model shared. Top1:67.408/Top5:87.258☆48Jun 22, 2018Updated 7 years ago
- Lightweight Mat and imread()/imwrite()/imshow()☆82Jun 17, 2024Updated last year
- An implementation of parameter server framework in PyTorch RPC.☆12Nov 12, 2021Updated 4 years ago
- Code and information for face image quality assessment with SER-FIQ☆577Dec 9, 2022Updated 3 years ago
- This repo contains code for *FD-MobileNet: Improved MobileNet with A Fast Downsampling Strategy*.☆35May 3, 2018Updated 7 years ago
- Pytorch implementation of CVPR2021 paper: SuperMix: Supervising the Mixing Data Augmentation☆92Jan 6, 2022Updated 4 years ago
- PyTorch Code for the Paper: "Exploiting Uncertainty of Loss Landscape for Stochastic Optimization [Bhaskara et al. (2019)]☆16Dec 8, 2025Updated 3 months ago
- A MXNet implementation of IBN-Net (Resnet-IBN-a, Resnext-IBN-a, Resnext-IBN-b)☆35Sep 28, 2018Updated 7 years ago
- Info how to profile multithreaded Python programs☆35Jul 18, 2017Updated 8 years ago
- A PyTorch implementation of adversarial pose estimation for multi-person☆66Jan 24, 2018Updated 8 years ago
- Reproduction of MobileNetV2 using MXNet☆128Mar 15, 2019Updated 6 years ago
- A MXNet implementation of Modified CRUNet & Residual Attention Network☆67Oct 22, 2018Updated 7 years ago
- SplitNet implemented based on ResNet-50 trained on ImageNet-22K☆16Jun 18, 2018Updated 7 years ago
- MB-LBP based face detection☆13Jul 1, 2021Updated 4 years ago
- A prototype for distributed training/validation/evaluation/extraction with PyTorch.☆14Jun 13, 2020Updated 5 years ago
- Implementation of ShuffleNetV2 for pytorch☆195Apr 17, 2022Updated 3 years ago
- Noise-Tolerant Paradigm for Training Face Recognition CNNs [Official, CVPR 2019]☆136Mar 27, 2019Updated 6 years ago
- Multi-modal data augmentation for machine learning☆16Jun 4, 2019Updated 6 years ago
- This is the code of the ICASSP 2020 paper "Joint phoneme alignment and text-informed speech separation on highly corrupted speech"☆15Apr 8, 2024Updated last year