We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Please consider replacing
img_batch, label_batch = dataiter.next()
with
img_batch, label_batch = next(dataiter)