프리트레인 질문있습니다
277
작성한 질문수 1
안녕하세요 이미지넷은 1000개의 클래스인걸로 알고있는데 model=models.resnet18(pretraine=True)하면 모델에서 어느 부분만 가져와서 어느부분 weight를 채워주는건가요? 이번 예시는 2개의 traget 값인데요
답변 1
0
안녕하세요, 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
감사합니다.
[개정판] 딥러닝 컴퓨터 비전 완벽 가이드 먼저? 구현하며 배우는 Transformer 먼저?
0
25
1
전 강의와 전혀 이어지지가 않음
0
26
1
Continual Learning 과 Transfer Learning 의 차이점
0
1473
1
Deep Learning 정의에 나온 Graphical representation learning에 대해서
0
602
1
학습시간 줄이는 방법에 대하여 문의 드리겠습니다.
0
1195
3
cross_entropy
0
2401
1
Mnist 데이터 실습 관련 질문입니다!
0
301
1
CNN_MNIST 실습예제 질문입니다.
0
260
1
ResNet 클래스의 _make_layer 메서드 부분의 설명이 이해가 되지 않습니다
0
761
2
CNN(강의자료 38 39 페이지 질문)
0
218
1
forward에서 h_t
0
219
1
[실습] MLP를 이용한 MNIST 숫자분류 - 테스트 데이터 셋에 라벨 제거
1
242
1
[실습] MLP를 이용한 MNIST 숫자분류 - 테스트 데이터 셋에 라벨이 붙어있어요
0
410
0
[실습] MLP를 이용한 MNIST 숫자분류 - 테스트 데이터 셋에 라벨이 붙어있어요
0
215
1
[실습] MLP를 이용한 MNIST 숫자분류 - 형태가 달라요
1
476
3
코드 오류
0
256
1
RNN 실습_ cuda 관련 질문 드립니다!
0
303
1
torchtext
0
675
3
LSTM 원리가 궁금합니다.
0
342
2
Autoencoder 계산
0
205
1
Pytorch 실습 진입장벽
0
238
1
AutoEncoder 질문 드립니다.
0
341
1
"딥러닝"이라는 제목의 강의 슬라이드 15페이지
0
184
1
DataLoader에서 num_workers 개념 설명 다시 한 번 부탁드려요
0
4994
1