Prenez le premier dans DatalOader Pytorch
inputs, classes = next(iter(dataloader))
Sleepy Shark
inputs, classes = next(iter(dataloader))