tczhangzhi / pytorch-parallelLinks
Optimize an example model with Python, CPP, and CUDA extensions and Ring-Allreduce.
☆110Updated 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源码阅读 0.2.0 版本☆91Updated 5 years ago
- Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation☆412Updated 3 years ago
- A super light-weight deep learning library based on NumPy in PyTorch fashion.☆94Updated 4 years ago
- Code for “Discrimination-aware-Channel-Pruning-for-Deep-Neural-Networks”☆183Updated 4 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆77Updated 6 years ago
- A hyperparameter manager for deep learning experiments.☆96Updated 2 years ago
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆84Updated 5 years ago
- An example of C++ extension for PyTorch.☆37Updated 6 years ago
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆167Updated 4 years ago
- A bite of cpp api in PyTorch-1.0☆163Updated 5 years ago
- Caffe implementation of ICCV 2017 & TPAMI 2018 paper - ThiNet☆46Updated 7 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
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆603Updated 5 years ago
- SPOS(Single Path One-Shot Neural Architecture Search with Uniform Sampling) rebuilt in Pytorch with single GPU.☆245Updated 3 years ago
- deformable_conv2d layer implemented in pytorch☆63Updated 6 years ago
- ☆39Updated 7 years ago
- ☆120Updated 5 years ago
- A Simple & Flexible Cross Framework Operators Toolkit☆164Updated 5 years ago
- [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy☆159Updated 5 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- Hands-on Tutorial on Automated Deep Learning☆150Updated 5 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆122Updated 6 years ago
- 基于旷视研究院领先的深度学习算法，提供满足多业务场景的预训练模型☆92Updated last year
- ☆263Updated 5 years ago
- Easy to use Pytorch☆70Updated 6 months ago
- How to write cuda kernels or c functions in pytorch, especially for former caffe users.☆54Updated 7 years ago
- This repo contains the official Pytorch reimplementation of the paper "NetAdapt: Platform-Aware Neural Network Adaptation for Mobile Appl…☆188Updated 2 years ago
- Neural architecture search(NAS)☆14Updated 6 years ago
- CNN channel pruning, LeGR, MorphNet, AMC. Codebase for paper "LeGR: Filter Pruning via Learned Global Ranking"☆114Updated 5 years ago
- ☆165Updated 2 years ago