안녕하세요 이미지넷은 1000개의 클래스인걸로 알고있는데 model=models.resnet18(pretraine=True)하면 모델에서 어느 부분만 가져와서 어느부분 weight를 채워주는건가요? 이번 예시는 2개의 traget 값인데요

안녕하세요, Justin입니다.
이미지넷은 말씀해주신 대로 1,000개의 클래스로 구성된 데이터셋입니다. model = models.resnet18(pretrain=True)로 설정하면 1,000개의 클래스로 학습이 된 ResNet18 모델 전체를 정의하게 되는 것이구요, 그 이후, 클래스 2개로 설정하기 위해 num_ftrs 부분과 model.fc 부분을 추가로 정의하여 학습을 진행하게 됩니다.&nbsp;
예제에 대한 도움을 드리기 위해, 관련된 링크를 첨부해드립니다.
https://tutorials.pytorch.kr/beginner/transfer_learning_tutorial.html
감사합니다.

인프런 커뮤니티 질문&답변

프리트레인 질문있습니다