본문 바로가기
Pytorch

requires_grad

by jun0823 2022. 4. 29.
반응형
for param in vgg16.features.parameters():
  param.requires_grad = False

모델에 있는 layer들을 for문을 이용해 가져와서 하나하나씩 false로 만들어 주는 코드

 

false로 만들어 주는 경우 파라미터들을 상수 취급 해주어서 training시 update가 되지 않도록 만들어준다.

반응형

댓글