tczhangzhi / pytorch-parallelLinks
Optimize an example model with Python, CPP, and CUDA extensions and Ring-Allreduce.
☆111Updated 6 years ago
Alternatives and similar repositories for pytorch-parallel
Users that are interested in pytorch-parallel are comparing it to the libraries listed below
Sorting:
- Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation☆412Updated 3 years ago
- pytorch源码阅读 0.2.0 版本☆91Updated 5 years ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆84Updated 5 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆604Updated 5 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- A Simple & Flexible Cross Framework Operators Toolkit☆164Updated 5 years ago
- A bite of cpp api in PyTorch-1.0☆163Updated 5 years ago
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆168Updated 4 years ago
- Hands-on Tutorial on Automated Deep Learning☆149Updated 5 years ago
- CUDA implementation of NMS for PyTorch☆88Updated 5 years ago
- A sample for onnxparser working with trt user defined plugins for TRT7.0☆169Updated 5 years ago
- Fast CUDA Kernels for ResNet Inference.☆180Updated 6 years ago
- study resources of model and engineering☆224Updated last year
- An example of C++ extension for PyTorch.☆37Updated 6 years ago
- Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"☆185Updated last year
- convert torch module to tensorrt network or tvm function☆89Updated 5 years ago
- A super light-weight deep learning library based on NumPy in PyTorch fashion.☆94Updated 4 years ago
- ☆131Updated 5 years ago
- Search for RegNet using PyTorch☆318Updated 5 years ago
- Code for “Discrimination-aware-Channel-Pruning-for-Deep-Neural-Networks”☆182Updated 5 years ago
- deformable_conv2d layer implemented in pytorch☆63Updated 6 years ago
- Caffe implementation of ICCV 2017 & TPAMI 2018 paper - ThiNet☆46Updated 7 years ago
- Benchmark of TVM quantized model on CUDA☆111Updated 5 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆77Updated 6 years ago
- Implementation of CVPR 2019 paper: Distilling Object Detectors with Fine-grained Feature Imitation☆419Updated 4 years ago
- A project demonstrate that downsampling(upsaming) in cnn are not nesscessary☆82Updated 5 years ago
- Personal Pytorch toy script.☆67Updated 3 years ago
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆198Updated 5 years ago
- ☆165Updated 2 years ago
- ☆150Updated 3 years ago