问题描述:
训练神经网络模型,训练时正常,训练一个epoch后测试的时候,报错RuntimeError: CUDA out of memory. 然而我的模型并没有很大,GPU占用率也不高。
解决方法:
with torch.no_grad():(your test code)
训练神经网络模型,训练时正常,训练一个epoch后测试的时候,报错RuntimeError: CUDA out of memory. 然而我的模型并没有很大,GPU占用率也不高。
with torch.no_grad():(your test code)