Dataset et DataLoader

Un Dataset expose les exemples de données (__len__, __getitem__) et un DataLoader les regroupe en lots (mini-batchs), avec mélange, chargement parallèle et gestion de la mémoire. L'entraînement par lots rend le calcul efficace sur GPU et réduit la consommation de mémoire.

Exemple

DataLoader(dataset, batch_size=64, shuffle=True)

Notions voisines

Tout le glossaire