Lyken17 / pytorch-memongerView external linksLinks
Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174
☆602Dec 27, 2019Updated 6 years ago
Alternatives and similar repositories for pytorch-memonger
Users that are interested in pytorch-memonger are comparing it to the libraries listed below
Sorting:
- My best practice of training large dataset using PyTorch.☆1,104May 9, 2024Updated last year
- Synchronized Batch Normalization implementation in PyTorch.☆1,503Apr 8, 2021Updated 4 years ago
- In-Place Activated BatchNorm for Memory-Optimized Training of DNNs☆1,335Jul 8, 2025Updated 7 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,918Updated this week
- Model analyzer in PyTorch☆1,502Mar 19, 2023Updated 2 years ago
- PyTorch layer-by-layer model profiler☆607May 23, 2021Updated 4 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- Some tricks of pytorch...☆1,194Jun 20, 2024Updated last year
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- pytorch memory track code☆1,017May 4, 2021Updated 4 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Jul 4, 2020Updated 5 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆635Jul 29, 2025Updated 6 months ago
- ResNeSt: Split-Attention Networks☆3,265Dec 9, 2022Updated 3 years ago
- ☆181Mar 3, 2023Updated 2 years ago
- Search for RegNet using PyTorch☆316May 30, 2020Updated 5 years ago
- ☆150Jul 18, 2020Updated 5 years ago
- TVM integration into PyTorch☆456Jan 15, 2020Updated 6 years ago
- Code for ICCV2019 paper "InstaBoost: Boosting Instance Segmentation Via Probability Map Guided Copy-Pasting"☆403Jan 7, 2020Updated 6 years ago
- ☆1,510Aug 27, 2020Updated 5 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 5 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆300Jun 7, 2025Updated 8 months ago
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆5,116Feb 3, 2026Updated last week
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆871Dec 8, 2020Updated 5 years ago
- PyTorch API for GluonCV Models☆531Nov 2, 2022Updated 3 years ago
- TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision☆2,256Nov 19, 2020Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆314Dec 18, 2019Updated 6 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,116Apr 22, 2022Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆366Apr 23, 2018Updated 7 years ago
- Debug PyTorch code using PySnooper☆800Apr 28, 2021Updated 4 years ago
- Pytorch implementation of newly added convolution☆585Oct 26, 2019Updated 6 years ago
- Code for Switchable Normalization from "Differentiable Learning-to-Normalize via Switchable Normalization", https://arxiv.org/abs/1806.10…☆870Jun 11, 2020Updated 5 years ago
- pip install antialiased-cnns to improve stability and accuracy☆1,680Apr 8, 2024Updated last year
- Densely Connected Search Space for More Flexible Neural Architecture Search (CVPR2020)☆294Jul 4, 2020Updated 5 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,169Mar 20, 2024Updated last year
- Instance-Batch Normalization Networks (ECCV2018)☆808Jul 4, 2023Updated 2 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,619Feb 8, 2026Updated last week
- End-to-End Object Detection with Fully Convolutional Network☆496Jan 10, 2022Updated 4 years ago
- [IROS 2021] ADD: A Fine-grained Dynamic Inference Architecture for Semantic Image Segmentation☆10May 3, 2022Updated 3 years ago
- An easy to understand and better performance version of CenterNet☆551Apr 12, 2021Updated 4 years ago