prigoyal / pytorch_memongerLinks
Experimental ground for optimizing memory of pytorch models
☆366Updated 7 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:
- PyTorch layer-by-layer model profiler☆607Updated 4 years ago
- Tutorial for building a custom CUDA function for Pytorch☆516Updated 6 years ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- Estimates the size of a PyTorch model in memory☆357Updated 5 years ago
- Implementation of https://arxiv.org/abs/1904.00962