Not enough VRAM after ~3 hours of training.
Hi Guys.
Code: Select all
I have an Nvidia 1650
After achieving some decent results following the tutorial to the letter, i tried my first train with Dlight. I tried with a batch-size of 12 and it crashed after about 3 hours due to not enough VRAM. I had no other program running.
After that, i tried with a batch-size of 10, in the same conditions, but the same thing is happening.
I am running the nvidia-smi and during the batch-size 10 training it shows a constant 2886 MiB out of 4096 MiB.
Am i doing something wrong?