Thanks .I solve it.
can you help me about this error:
RuntimeError: CUDA out of memory. Tried to allocate 192.00 MiB (GPU 0; 15.90 GiB total capacity; 14.80 GiB already allocated; 77.06 MiB free; 15.01 GiB reserved in total by PyTorch)
I used:
import gc
gc.collect()
and
torch.cuda.empty_cache()
but I also got the error.