RuntimeError: CUDA out of memory. Tried to allocate 4.29 GiB (GPU 0; 16.00 GiB total capacity; 504.24 MiB already allocated; 13.20 GiB free; 758.00 MiB reserved in total by PyTorch)

 

 

 

#############################