From fd4e40ae2ae58c06226cc9eb4c2ae9bdcfb677fd Mon Sep 17 00:00:00 2001 From: zhang Date: Sun, 31 Jul 2022 13:07:02 +0800 Subject: wordpiece --- projs/01-fashion-mnist/00_dataset_dataloader.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 projs/01-fashion-mnist/00_dataset_dataloader.py (limited to 'projs/01-fashion-mnist') 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 -- cgit v1.2.3