chenyf2022

4090显卡CUDA out of memory

2

torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 4.56 GiB (GPU 0; 23.68 GiB total capacity; 18.96 GiB already allocated; 2.84 GiB free; 18.97 GiB reserved in total by PyTorch)...

Hello author, I would like to ask whether this model can run on the 4090 machine? When I tried to reproduce this paper, there was a bug that I can't...

请问可以告知您微调数据集是哪一份吗？