超分辨率重建复现测试数据集gpu显示内存不够怎么办

试了几个办法比如把批量大小从128改到32，换更小的数据集都是显示这个要用3.5g内存，这内存到底和啥有关啊，
问了chatgpt哪个方法还能用啊

是 GPU 内存不够吧？如果是训练的话，4G内存的显卡太小了。