tczhangzhi / pytorch-parallel
Optimize an example model with Python, CPP, and CUDA extensions and Ring-Allreduce.
☆108Updated 5 years ago
Related projects: ⓘ
- Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation☆409Updated 2 years ago
- pytorch源码阅读 0.2.0 版本☆88Updated 4 years ago
- Implementation of CVPR 2019 paper: Distilling Object Detectors with Fine-grained Feature Imitation☆413Updated 3 years ago
- ☆130Updated 3 years ago
- ☆118Updated this week
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆79Updated 4 years ago
- ☆128Updated this week
- An example of C++ extension for PyTorch.☆36Updated 4 years ago
- A small deep-learning framework with C++/Python/CUDA☆53Updated 6 years ago
- [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices☆165Updated 3 years ago
- MobileNetV3, PyTorch, pretrained model, on ImageNet 1K, detailed training processing.☆56Updated 5 years ago
- CUDA implementation of NMS for PyTorch☆86Updated 4 years ago
- ☆60Updated this week
- Tensorflow implement of "SlowFast Networks for Video Recognition"☆46Updated 5 years ago
- Easy Panoptic Segmentation implement in mmdet☆11Updated 5 years ago
- Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”☆76Updated 5 years ago
- A way to use cuda to accelerate top k algorithm☆29Updated 7 years ago
- ☆119Updated 4 years ago
- A super light-weight deep learning library based on NumPy in PyTorch fashion.☆91Updated 2 years ago
- Convert image folder to lmdb, adapted from Efficient-PyTorch☆66Updated last year
- ☆107Updated this week
- PyTorch Dataset Rank Dataset☆38Updated 3 years ago
- Deep Learning Accelerate Knowledge Review☆33Updated 5 years ago
- deformable_conv2d layer implemented in pytorch☆64Updated 5 years ago
- The MXNet Implementation of Enhanced SSH (ESSH) for Face Detection and Alignment☆55Updated 5 years ago
- 分享计算机视觉每天的arXiv文章☆711Updated 5 years ago
- Personal Pytorch toy script.☆66Updated 2 years ago
- alibabacloud-quantization-networks☆121Updated 4 years ago
- Single Path One-Shot NAS MXNet implementation with full training and searching pipeline. Support both Block and Channel Selection. Search…☆151Updated 4 years ago
- ☆165Updated last year