rentruewang/koila: Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
Category: ml
Language: Python
Topics: deep-learning, gradient-accumulation, lazy-evaluation, machine-learning, memory-management, neural-network, out-of-memory, pytorch
Homepage: https://rentruewang.github.io/koila/