어떤가요? 앞에서 정의했던 vgg11_layers, 완전연결층과 출력층((classifier): Sequential()) 부분을 합친 것을 확인할 수 있습니다.

VGG11 모델을 사용하는 것이 어렵게 느껴지나요? 그렇다면 VGG11을 더 쉽게 사용할 수 있는 방법이 있습니다. 사실 VGG 모델은 사전 훈련된 모델입니다. 이미 누군가가 대용량의 이미지 데이터로 학습을 시켰으며, 최상의 상태로 튜닝을 거쳐 모든 사람이 사용할 수 있도록 공유한 사전 훈련된 모델입니다. 따라서 다음과 같이 코드 한두 줄로 쉽게 가져다 사용할 수 있습니다.

코드 6-46 VGG11 사전 훈련된 모델 사용

import torchvision.models as models
pretrained_model = models.vgg11_bn(pretrained=True) ------ ①
print(pretrained_model)

① 배치 정규화가 적용된 사전 훈련된 VGG11 모델을 사용하기 위해서는 다음과 같은 파라미터를 사용합니다.

ⓐ vgg11_bn은 VGG11 기본 모델에 배치 정규화가 적용된 모델을 사용하겠다는 의미입니다.

ⓑ pretrained를 True로 설정하면 사전 훈련된 모델을 사용(미리 학습된 파라미터 값들을 사용)하겠다는 의미입니다.