Obtenez-en un de DatalOader

 inputs, classes = next(iter(dataloader))   
Sleepy Shark