CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0; 15.90 GiB total capacity; 14.80 GiB already allocated; 47.88 MiB free; 15.16 GiB reserved in total by PyTorch)

[albanD]

Hi,

Not much, that’s why I did not answer anything
You can check this related post for potential solutions: How can you train your model on large batches when your GPUs can only hold couple of batches