标签： pytorch
当我用512*512用batch_size=4图像训练模型时，它可以工作， 但是当我用1024*1024用batch_size=1图像训练它时，会发生cuda内存不足错误。
512*512
batch_size=4
1024*1024
batch_size=1
cuda
我很困惑，像素总数相等。为什么一张1024*1024图像比四张512*512图像占用更多的内存？