반응형 Pytorch1 requires_grad for param in vgg16.features.parameters(): param.requires_grad = False 모델에 있는 layer들을 for문을 이용해 가져와서 하나하나씩 false로 만들어 주는 코드 false로 만들어 주는 경우 파라미터들을 상수 취급 해주어서 training시 update가 되지 않도록 만들어준다. 2022. 4. 29. 이전 1 다음 반응형