RuntimeError: CUDA out of memory. Tried to allocate 15.94 GiB (GPU 0; 15.78 GiB total capacity; 641.30 MiB already allocated; 13.86 GiB free; 642.00 MiB reserved in total by PyTorch)
i got this error when i  train model