Obtenga un lote único del cargador de datos de la antorcha
inputs, classes = next(iter(dataloader))
Difficult Dormouse
inputs, classes = next(iter(dataloader))