Dataset en DataLoader met twee uitgangen

In deze en de volgende oefeningen bouw je een model met twee uitgangen om zowel het teken als het alfabet waarvan het afkomstig is te voorspellen op basis van de afbeelding van het teken. Zoals altijd begin je met het voorbereiden van de data.

De klasse OmniglotDataset die je eerder hebt gemaakt is beschikbaar, samen met bijgewerkte samples. Gebruik deze om de Dataset en de DataLoader te bouwen.

De volgende imports zijn al voor je gedaan:

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