运行main.py文件，报错内存不够是什么原因，可以修改什么？谢谢 #12
Description
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 4.08 GiB already allocated; 10.56 MiB free; 4.18 GiB reserved in total by PyTorch)
{{ message }}
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 4.08 GiB already allocated; 10.56 MiB free; 4.18 GiB reserved in total by PyTorch)