Skip to content

Instantly share code, notes, and snippets.

@SubhadityaMukherjee

SubhadityaMukherjee/ds.md

Created February 13, 2020 06:43
Show Gist options
  • Save SubhadityaMukherjee/c9f8a0479b99d2df57b659b5800bda78 to your computer and use it in GitHub Desktop.
Save SubhadityaMukherjee/c9f8a0479b99d2df57b659b5800bda78 to your computer and use it in GitHub Desktop.
Download ZIP
ds
Raw
ds.md 
dataset = dset.ImageFolder(root=dataroot,
                           transform=transforms.Compose([
                               transforms.Resize(image_size),
                               transforms.CenterCrop(image_size),
                               transforms.ToTensor(),
                               transforms.Normalize((.5, .5, .5),
                                                    (.5, .5, .5)),
                           ]))
dataloader = torch.utils.data.DataLoader(dataset,
                                         batch_size=batch_size,
                                         shuffle=True,
                                         num_workers=workers)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment