inplace_abn
Memory optimizer
An optimization technique to reduce memory usage in deep neural networks during training
In-Place Activated BatchNorm for Memory-Optimized Training of DNNs
1k stars
39 watching
187 forks
Language: Python
last commit: 3 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 dmlc/mxnet-memonger | A tool for optimizing deep learning models to reduce memory usage without sacrificing performance | 308 |
 fmassa/optimize-net | An optimization library for reducing memory usage in PyTorch neural networks | 282 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,821 |
 xternalz/sdpoint | A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency. | 18 |
 brml/climin | A framework for optimizing machine learning functions using gradient-based optimization methods. | 180 |
 neuralmagic/sparseml | Enables the creation of smaller neural network models through efficient pruning and quantization techniques | 2,071 |
 atgambardella/pytorch-es | An implementation of an optimization algorithm for training neural networks in machine learning environments. | 351 |
 datacanvasio/hypergbm | An AutoML toolkit designed to automate the entire machine learning process pipeline for tabular data | 337 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 132 |
 delta2323/gb-gnn | Analyzes and optimizes the performance of graph neural networks using gradient boosting and various aggregation models. | 13 |
 typemonkey/turtle | A programming language designed to optimize memory usage through long-term storage of data | 5 |
 khanrc/pt.darts | An implementation of DARTS, a method for automatically designing neural network architectures. | 441 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 513 |
 mengomarlene/opti4abq | An optimisation method that minimises the difference between FEA output and data in Abaqus models | 16 |
 cn-upb/deepcomp | A reinforcement learning-based system for optimizing multi-cell selection in wireless networks | 58 |