Goto

Collaborating Authors

 cuda error


GitHub - rentruewang/koila: Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

#artificialintelligence

Koila solves CUDA error: out of memory error painlessly. Fix it with just one line of code, and forget it. To use it, download release v0.1.1 here.* Automatically accumulates gradients when batch sizes are too large. Lazily evaluates PyTorch code to save computing power.