Hi
I’m using mnist dataset in my code , but there’s an error when i run it
Traceback (most recent call last):
  File "C:\Users\Ati\Desktop\GD\GD.py", line 306, in <module>
    update_interval=update_interval, max_prob=max_prob, plot=plot, train=train, gpu=gpu)
  File "C:\Users\Ati\Desktop\GD\GD.py", line 105, in main
    images, labels = dataiter.next()
ValueError: too many values to unpack (expected 2)

i think problem is related to this part of my code:
 transform = transforms.Compose([transforms.ToTensor(),
                              transforms.Normalize((0.5,), (0.5,)),
                              ])
    root = data_path
    dataset = MNIST(root='root', download=True , transform=transform)

    trainloader = torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=True)
    dataiter = iter(trainloader)
    images, labels = dataiter.next()
    images, labels = images.view(-1, 784) ,  labels

Could someone give me a solution for this problem please?