-
질문 & 답변
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
프리트레인 질문있습니다
20.12.29 19:14 작성 조회수 130
0
안녕하세요 이미지넷은 1000개의 클래스인걸로 알고있는데 model=models.resnet18(pretraine=True)하면 모델에서 어느 부분만 가져와서 어느부분 weight를 채워주는건가요? 이번 예시는 2개의 traget 값인데요
답변을 작성해보세요.
0
Justin
지식공유자2020.12.29
안녕하세요, Justin입니다.
이미지넷은 말씀해주신 대로 1,000개의 클래스로 구성된 데이터셋입니다. model = models.resnet18(pretrain=True)로 설정하면 1,000개의 클래스로 학습이 된 ResNet18 모델 전체를 정의하게 되는 것이구요, 그 이후, 클래스 2개로 설정하기 위해 num_ftrs 부분과 model.fc 부분을 추가로 정의하여 학습을 진행하게 됩니다.
예제에 대한 도움을 드리기 위해, 관련된 링크를 첨부해드립니다.
https://tutorials.pytorch.kr/beginner/transfer_learning_tutorial.html
감사합니다.
답변 1