SOLVED - IndexError: too many indices for tensor of dimension 0 from DCGAN tutorial

[seyonechithrananda]

IndexError: Traceback (most recent call last): File "/Applications/anaconda3/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/Applications/anaconda3/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 99, in <listcomp> samples = collate_fn([dataset[i] for i in batch_indices]) File "/Applications/anaconda3/lib/python3.7/site-packages/torchvision/datasets/folder.py", line 140, in __getitem__ sample = self.transform(sample) File "/Applications/anaconda3/lib/python3.7/site-packages/torchvision/transforms/transforms.py", line 61, in __call__ img = t(img) File "/Applications/anaconda3/lib/python3.7/site-packages/torchvision/transforms/transforms.py", line 164, in __call__ return F.normalize(tensor, self.mean, self.std, self.inplace) File "/Applications/anaconda3/lib/python3.7/site-packages/torchvision/transforms/functional.py", line 208, in normalize tensor.sub_(mean[:, None, None]).div_(std[:, None, None])`
IndexError: too many indices for tensor of dimension 0

I get this error when trying to load images from the following path with the CelebA image dataset using ImageFolder:
dataroot = "/Users/seyonechithrananda/dcgan/data"
dataset = dset.ImageFolder(root=dataroot, transform = transforms.Compose([ transforms.Resize(image_size), transforms.CenterCrop(image_size), transforms.ToTensor(), transforms.Normalize(0.5, 0.5, 0.5), (0.5, 0.5, 0.5) ]))
dataloader = torch.utils.data.DataLoader(dataset, batch_size = batch_size, shuffle=True, num_workers = workers)
device = torch.device("cpu")
real_batch = next(iter(dataloader))
plt.figure(figsize = (8,8))
plt.axis('off')
plt.title("Training Images")
plt.imshow(np.transpose(vutils.make_grid(real_batch[0].to(device)[:64], padding=2, normalize=True).cpu(),(1 ,2,0)))

[seyonechithrananda]

This issue can be solve by changing the parameters for transforms.Normalize from (0.5, 0.5, 0.5), (0.5, 0.5, 0.5) to (0.5,), (0.5)

[ghltshubh]

@seyonechithrananda Can you please make an amendment from (0.5,), (0.5) to (0.5,), (0.5,)