美文网首页
(16)验证集上内存溢出问题

(16)验证集上内存溢出问题

作者: 顽皮的石头7788121 | 来源:发表于2019-05-05 22:32 被阅读0次

在训练过程中一直运行良好,但是在验证集上测试的时候内存溢出,GPU显存其实很大,batch size 也不是很大。通过

    with torch.no_grad():
        for ii, data in enumerate(dataloader):
            input, label = data

解决
网上分析可能是梯度在验证阶段不需要回传造成没有被释放并累积造成的。

相关文章

网友评论

      本文标题:(16)验证集上内存溢出问题

      本文链接:https://www.haomeiwen.com/subject/pkbkoqtx.html