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

안녕하세요, Justin입니다.

문의주신 내용을 확인해보았을 때, 08_Transfer_Learning.ipynb 코드 내용으로 파악되는데 맞나요?

해당 스크립트는 ImageNET 데이터를 이용하여 학습이 완료된 ResNet18 모델을 이용하여 특정 도메인의 데이터를 학습하는 실습내용입니다. 본 예제에서는 2개의 클래스로 구성된 이미지 데이터를 이용하여 적용해보는 내용입니다.

코드 실행시 발생한 에러는 보통 모델의 출력값과 비교하는 레이블의 크기가 맞지 않아서 발생되는 에러이므로, 학습이 진행될 때, 입력받는 데이터와 레이블이 올바르게 구성되어있는지 확인해보시길 바랍니다.

[확인해야 하는 코드]

image_datasets = {x: datasets.ImageFolder("../data/hymenoptera_data", data_transforms[x]) for x in ['train', 'val']}

dataloaders = {x: torch.utils.data.DaataLoader(image_datasets[x], batch_size = 8, num_workers = 0, shuffle = True) for x in ['train', 'val']}

감사합니다.