Pytorch) Out of memory error during evaluation

Training epoch은 잘 돌았는데 evaluation epoch에서 cuda: out of memory 에러가 나는 경우

💡
net.eval() 시 with torch.no_grad(): 꼭 붙이기
torch.cuda.empty_cache() 여기저기 추가해주기