Получите одиночную партию от загрузчика данных Torch
inputs, classes = next(iter(dataloader))
Difficult Dormouse
inputs, classes = next(iter(dataloader))