pytorch-MNIST-CelebA-cGAN-cDCGAN icon indicating copy to clipboard operation
pytorch-MNIST-CelebA-cGAN-cDCGAN copied to clipboard

Published 20 hours ago • verified publisher icon znxlwm

How to pass custom dataset?

Open hinash88 opened this issue 5 years ago • 0 comments

Can Anybody please help how to pass custom datasets instead of MNIST?

train_loader = torch.utils.data.DataLoader( datasets.MNIST('data', train=True, download=False, transform=transform), batch_size=batch_size, shuffle=True)

hinash88 avatar Mar 26 '19 05:03 hinash88