and it’s showing me this error now when i minimised the data size :
RuntimeError: CUDA out of memory. Tried to allocate 2.00 GiB (GPU 0; 11.17 GiB total capacity; 8.99 GiB already allocated; 1.32 GiB free; 9.39 GiB reserved in total by PyTorch)