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가 되지 않도록 만들어준다.
반응형