koila
Koila是一个开源项目,通过一行代码解决CUDA内存不足错误。它包含自动累积梯度、惰性求值和批次大小优化功能,使GPU计算更高效。作为轻量级PyTorch包装器,Koila自动计算GPU的剩余内存,并选用适当的批次大小,从而保持与PyTorch的兼容性。