https://pytorch.org/docs/stable/checkpoint.html torch.utils.checkpoint — PyTorch 1.13 documentation Shortcuts pytorch.org 많은 수의 파라미터를 가지는 크기가 큰 딥러닝 모델의 경우, batch를 1로 설정하더라도 'out of memory'를 만나게 될 수 있다. 나의 경우, Transformer backbone을 사용하는 VSR 모델에 대한 실험을 해보려하니 48GB 메모리를 가지는 Quadro RTX 8000 GPU로도 실험이 불가능하였다. GPU를 활용한 실험을 할 때 이러한 메모리 제한 문제는 'torch.utils.checkpoint'로 해결 할 수 있다. tutorial을 보면, 아래와 같이 쓰여있..