GitHub
  
  
    

pytorch/tutorials 29

PyTorch tutorials. Contribute to pytorch/tutorials development by creating an account on GitHub.


  
  
    
    
  
  


In the above tutorial, there is a code as below.
Does this code load only mini-batch data from the disk after running “dataiter.next()”?
Or, does this code load the entire dataset from the disk to the memory ?
transform=transforms.Compose([transforms.ToTensor(),
                              transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
                             ])
trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, 
                                          shuffle=True, num_workers=2)

testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=4, 
                                          shuffle=False, num_workers=2)

dataiter = iter(trainloader)
images, labels = dataiter.next()