PyTorch 1.13.1+cpu next(iter(training_loader)) genera TypeError

De esta respuesta, sé que desde PyTorch 1.13, la sintaxis del iterador se convierte en:

next(iter(training_loader))

Mi código original es:

dataiter = iter(training_loader)
images, labels = dataiter.next()
img_grid = torchvision.utils.make_grid(images)

Luego cambié los códigos a esto:

images, labels = next(iter(training_loader))
img_grid = torchvision.utils.make_grid(images)

pero luego ocurre este error:

TypeError: tensor o lista de tensores esperados, obtuve

Pregunta: ¿Cómo debo hacer los cambios para borrar el error?

Usando PyTorch 1.13.1+cpu