RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 6.00 GiB total capacity; 229.16 MiB already allocated; 0 bytes free; 4.47 GiB reserved in total by PyTorch)
老师您好,我已经在自己的数据集上跑通了S和M,现在在尝试nano和tiny,但是尽管我调小了batchsize到2,依然会一直报上述错误,我想知道该怎么解决,我在跑S时候用的batch是8,m时候用的4。