torch.no_grad()는 뭘 해주는 함수인가?

파이토치 문서의 torch.no_grad() 설명

 

간단 정리

• gradient 연산을 옵션을 끌 때 사용하는 파이썬 컨텍스트 매니저
• 이 컨텍스트 내부에서 새로 생성된 텐서들은 requires_grad=False 상태가 되어, 메모리 사용량을 아껴준다.
• 파이썬의 데코레이터로도 사용할 수 있다