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