ML Framework (1) 썸네일형 리스트형 [PyTorch] torch.no_grad()와 model.eval()의 차이 👋 Intro 안녕하세요, 유블린입니다. 이 글에서는 torch.no_grad()와 model.eval()의 차이점을 다뤄보려고 합니다. 머신러닝 프레임 워크인 Pytorch로 모델 학습을 진행한 뒤 모델의 Evaluation을 진행 할 때 아래 처럼 model.eval()을 했는데 torch.no_grad()을 또 사용 하는 경우를 보셨나요? 혹은, 모델 학습은 잘 진행 했는데 Inference를 하는 중에 Memory leak이 발생한 적이 있나요? 후자의 경우에도 해결책 중 하나로 아래 코드를 제시 합니다. 왜 그럴까요? model.eval() with torch.no_grad(): for batch in data_loader: 여기서 저와 같은 의문을 가지셨다면, 잘 찾아오셨습니다. 저 또한 처음.. 이전 1 다음
