Obtenga un lote único del cargador de datos de la antorcha

inputs, classes = next(iter(dataloader))
Difficult Dormouse