I am getting the following error. I am using batch size 1.
I was able to train the very same network several times with batch size 6. the code does not show any error untill i use model.cuda(). Before i use model.cuda() the gpu memory fills up with 10gb.
RuntimeError: CUDA out of memory. Tried to allocate 10.24 GiB (GPU 3; 14.76 GiB total capacity; 10.31 GiB already allocated; 4.04 GiB free; 10.52 MiB cached)
When i first trained my model it used 10gb but now tries to use 10more gb.