fast-reid icon indicating copy to clipboard operation
fast-reid copied to clipboard

显存增长问题

Open annyhou opened this issue 3 years ago • 0 comments

image

问题描述:项目:FastRetri,backbone:resnet50,训练测试batchsize:32,第一次验证,显存暴涨至7G多,后面训练验证能稳定到4G以内。由于最大显存为7G多,相当于我需要提供的显存要在8G以上。为什么第一次验证时,需要那么大的显存缓冲区空间?可以减小么?

annyhou avatar Sep 20 '22 09:09 annyhou