Pytorch 测试阶段显存溢出

Pytorch模型训练的测试阶段, 显存溢出，训练阶段并没有显存溢出，原因:  没有使用 with torch.no_grad()，model.eval()只是使dropout和bn不工作，并不能产生torch.no_grad()的效果

with torch.no_grad():
    测试阶段代码