Click here to Skip to main content
15,868,016 members
Please Sign up or sign in to vote.
1.00/5 (1 vote)
See more:
Python
transform = transforms.Compose([transforms.ToTensor(),
                                transforms.Normalize(mean=[0.5],
                                std=[0.5])]
)

Python
mnist = datasets.MNIST(root='../data/', train=True, transform=transform, download=True)


Python
data_loader = DataLoader(dataset=mnist, batch_size=batch_size, shuffle=True, drop_last=True, pin_memory=True)


What I have tried:

<pre lang="Python">transform = transforms.Compose([transforms.ToTensor(),
                                transforms.Normalize(mean=[0.5],
                                std=[0.5])]
)

Python
mnis=(root='/content/drive/MyDrive/computer_vision/main_dataset5/train/', train=True, transform=transform, download=True)


Python
data_loader = DataLoader(dataset=mnist, batch_size=batch_size, shuffle=True, drop_last=True, pin_memory=True)
Posted
Updated 18-Feb-23 3:06am
Comments
Richard MacCutchan 18-Feb-23 9:43am    
What is the problem?
Member 12180276 18-Feb-23 10:03am    
unable to load the data
Dave Kreskowiak 18-Feb-23 12:01pm    
...and the error message would be...? (Hint: very important!)

This content, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)



CodeProject, 20 Bay Street, 11th Floor Toronto, Ontario, Canada M5J 2N8 +1 (416) 849-8900