Question about CUDA out of memory

[slptongji]

I run the code on a 10 GB GPU but it seems to have the problem of CUDA out of memory. I tried to reduce the batch_size to 1 but it still doesn't work.

[yahiko-l]

In fact, you should replace it with an accelerated device with more memory, such as a 3090 or 4090.