[Torch] 사전 훈련된 모델 불러오기, 학습 여부 조정

 

import torchvision.models as models

# 무작위 weight로 초기화된 모델
resnet18 = models.resnet18()
vgg16 = models.vgg16()

# 사전 훈련된 가중치로 초기화된 모델
resnet18 = models.resnet18(pretrained = True)
vgg16 = models.vgg16(pretrained = True)

# 모델 학습 여부 조정 - 가중치 고정
for param in resnet18.parameters():
	param.requires_grad = False

 

Reference

 

http://www.yes24.com/Product/Goods/107664335

딥러닝 파이토치 교과서 - YES24