rentruewang / koila

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
https://rentruewang.github.io/koila/
Apache License 2.0
1.82k stars 62 forks source link