기본적인 합성곱층을 사용하는 것도 복잡한데 아이덴티티 매핑, 다운샘플 및 병목 블록까지 정의하여 사용하려니 꽤 번거롭습니다.

ResNet 역시 사전 훈련된 모델이기 때문에 코드 한 줄로 간단하게 사용할 수 있습니다. 이제 그 방법에 대해 잠깐 살펴보겠습니다.

코드 6-79 사전 훈련된 ResNet 모델 사용

pretrained_model = models.resnet50(pretrained=True) ------ 사전 훈련된 ResNet 모델 사용을 위해서는 pretrained=True로 설정

우리가 만든 ResNet 네트워크와 사전 훈련된 ResNet 네트워크가 동일한지 확인해 봅시다.

코드 6-80 사전 훈련된 ResNet 네트워크 확인

print(pretrained_model)