tczhangzhi / pytorch-parallel

Related projects ⓘ

Alternatives and complementary repositories for pytorch-parallel

• MenghaoGuo / AutoDeeplab
  Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation
  ☆410Updated 2 years ago
• HadXu / Thunder
  A small deep-learning framework with C++/Python/CUDA
  ☆53Updated 6 years ago
• BIGKnight / deformable_conv2d_pytorch
  deformable_conv2d layer implemented in pytorch
  ☆62Updated 5 years ago
• keithyin / read-pytorch-source-code
  pytorch源码阅读 0.2.0 版本
  ☆88Updated 4 years ago
• iamhankai / cpp-pytorch
  A bite of cpp api in PyTorch-1.0
  ☆164Updated 4 years ago
• twangnh / Distilling-Object-Detectors
  Implementation of CVPR 2019 paper: Distilling Object Detectors with Fine-grained Feature Imitation
  ☆413Updated 3 years ago
• ZichenTian / DL_accelerate_review
  Deep Learning Accelerate Knowledge Review
  ☆33Updated 5 years ago
• mit-han-lab / amc-models
  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices
  ☆166Updated 3 years ago
• gdlg / pytorch_nms
  CUDA implementation of NMS for PyTorch
  ☆85Updated 4 years ago
• DingXiaoH / GSM-SGD
  Global Sparse Momentum SGD for pruning very deep neural networks
  ☆43Updated 2 years ago
• HRanWang / Spatial-Re-Scaling
  ☆130Updated 4 years ago
• sebgao / cTensor
  A super light-weight deep learning library based on NumPy in PyTorch fashion.
  ☆93Updated 3 years ago
• bindog / pytorch-model-parallel
  A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
  ☆82Updated 4 years ago
• Fangyh09 / Image2LMDB
  Convert image folder to lmdb, adapted from Efficient-PyTorch
  ☆66Updated last year
• yuxianzhi / Top-K
  A way to use cuda to accelerate top k algorithm
  ☆29Updated 7 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆129Updated last year
• BBuf / Memory-efficient-Convolution-for-Deep-Neural-Network
  ☆19Updated 4 years ago
• simochen / model-tools
  Tools for computing model parameters and FLOPs.
  ☆86Updated 5 years ago
• mengrang / Slimming-pytorch
  Pytorch Implementationg of “Learning Efficient Convolutional Networks through Network Slimming”
  ☆77Updated 5 years ago
• bindog / onnx-surgery
  ☆78Updated 4 years ago
• SCUT-AILab / DCP
  Code for “Discrimination-aware-Channel-Pruning-for-Deep-Neural-Networks”
  ☆184Updated 4 years ago
• Zx55 / dmcp
  ☆120Updated 4 years ago
• Lyken17 / pytorch-memonger
  Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174
  ☆589Updated 4 years ago
• pytorch / extension-script
  Example repository for custom C++/CUDA operators for TorchScript
  ☆114Updated 2 years ago
• TrojanXu / onnxparser-trt-plugin-sample
  A sample for onnxparser working with trt user defined plugins for TRT7.0
  ☆166Updated 4 years ago
• qinjian623 / pytorch_toys
  Personal Pytorch toy script.
  ☆67Updated 2 years ago
• zhengzhugithub / CV-arXiv-Daily
  分享计算机视觉每天的arXiv文章
  ☆711Updated 5 years ago