Obtenha um do Dataloader

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