search

rentruewang / koila

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

Getting NameError for lazy in (input, label) = lazy(input, label, batch=0) ? #34

Open Aayush-Jain01 opened 1 year ago

rentruewang commented 1 year ago

Sorry I somehow missed this issue, I'll take a look.