Pytorch

requires_grad

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

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

 

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

반응형