diff options
Diffstat (limited to 'projs')
| -rw-r--r-- | projs/01-fashion-mnist/00_dataset_dataloader.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/projs/01-fashion-mnist/00_dataset_dataloader.py b/projs/01-fashion-mnist/00_dataset_dataloader.py new file mode 100644 index 0000000..e967821 --- /dev/null +++ b/projs/01-fashion-mnist/00_dataset_dataloader.py @@ -0,0 +1,20 @@ + +from torch.utils.data import Dataset +from torchvision import datasets +from torchvision import transforms as T +import torch + +training_dataset = datasets.FashionMNIST(root='./data', train=True, transform=T.ToTensor(), download=True) +test_dataset = datasets.FashionMNIST(root='./data', train=False, transform=T.ToTensor(), download=True) + + +print(training_dataset.classes) + +training_loader = torch.utils.data.DataLoader(training_dataset, batch_size=4, shuffle=True, num_workers=0) +validation_loader = torch.utils.data.DataLoader(test_dataset, batch_size=4, shuffle=False, num_workers=0) + +# next(iter(training_loader)) + +for i, data in enumerate(training_loader): + batch_images, batch_labels = data + break |