最近使用GPU训练一个用PyTorch写的神经网络模型,网络比较深,有100层左右,训练部分的时候没什么问题,然而开始测试/验证环节的时候,虽然我用了torch.cuda.empty_cache()
,但是竟然直接报错:
https://www.jianshu.com/p/4a2be315b32a
预测时:
with torch.no_grad():
...
net=Net()
pred_for_validation=net(input)
...
最近使用GPU训练一个用PyTorch写的神经网络模型,网络比较深,有100层左右,训练部分的时候没什么问题,然而开始测试/验证环节的时候,虽然我用了torch.cuda.empty_cache()
,但是竟然直接报错:
https://www.jianshu.com/p/4a2be315b32a
预测时:
with torch.no_grad():
...
net=Net()
pred_for_validation=net(input)
...