use iter to get only one image at a time
data = iter(train_loader)

next(data) loads the one image and then unpacks into the variables image and labels
image,labels = next(data)
output = model(images)