在训练过程中一直运行良好,但是在验证集上测试的时候内存溢出,GPU显存其实很大,batch size 也不是很大。通过
with torch.no_grad():
for ii, data in enumerate(dataloader):
input, label = data
解决
网上分析可能是梯度在验证阶段不需要回传造成没有被释放并累积造成的。
在训练过程中一直运行良好,但是在验证集上测试的时候内存溢出,GPU显存其实很大,batch size 也不是很大。通过
with torch.no_grad():
for ii, data in enumerate(dataloader):
input, label = data
解决
网上分析可能是梯度在验证阶段不需要回传造成没有被释放并累积造成的。
本文标题:(16)验证集上内存溢出问题
本文链接:https://www.haomeiwen.com/subject/pkbkoqtx.html
网友评论