koila icon indicating copy to clipboard operation
koila copied to clipboard

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

Results 12 koila issues
Sort by recently updated
recently updated
newest added

I am using Koila to solve an OOM error during my training. But the following error occurs : ``Traceback (most recent call last): File "/mnt/sdb2/Adama/configure_docker_for_transvw/pytorch/train.py", line 92, in loss.backward() File...

I'm planning on making a major overhaul, to simplify the code and make it more scalable. Currently this project relies too much on checks to determine if an object is...

I'm still getting the memory error... wondering if it's because of conv layers or batch norm...

I am trying to apply koila lazy eval on a Unet3D. ``` # defining the model import torch import torch.nn as nn import torch.nn.functional as F def conv3(in_channels, out_channels, stride,...

I'm wondering what makes it incompatible with python 3.7 because that is the python version I am using and I can't upgrade to 3.8

Thanks for your nice work! I wrapped my input and label `(feat, label) = lazy(feat, label, batch=0)` Then I met the following error when running it. > File "/home/victor/anaconda3/envs/py38_tab/lib/python3.8/site-packages/koila/lazy.py", line...

Hello, I noticed you fix the lazy label bug and the getting-started.py is able to run. But it can not pass the assertion. The grad diff is quite large! assert...

If somehow we can integrate this with hugging face models while doing inference then its job is done for production-level deployments.

Thanks for your nice project. But it is not compatible with [einops](https://github.com/arogozhnikov/einops). Einops is convenient for pytorch user, and I think many people use it. I hope it works on...