Seonglae ChoMemory efficient (without back propagation)
validation inner loop를 이 context manager로 감싸서 gradient calculation을 방지
자동으로 gradient를 트래킹하지 않는다
@torch.no_grad()
no_grad — PyTorch 2.1 documentation
Community
https://pytorch.org/docs/stable/generated/torch.no_grad.html