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