두 개의 출력이 있는 Dataset과 DataLoader

이번 연습과 다음 연습에서는 문자 이미지를 보고 해당 문자가 무엇인지와 그 문자가 속한 알파벳을 동시에 예측하는, 두 개의 출력을 가지는 모델을 만들어 볼 거예요.いつものように, 먼저 데이터를 준비하는 것부터 시작합니다.

이전에 만들었던 OmniglotDataset 클래스를 업데이트된 samples와 함께 사용할 수 있어요. 이를 사용해 Dataset과 DataLoader를 구성해 봅시다.

아래 임포트는 미리 되어 있습니다:

from torch.utils.data import Dataset, DataLoader
from torchvision import transforms