search

corenel / torchzoo

Zoo of models and datasets for PyTorch.
MIT License
2 stars 3 forks source link

Error occur during loading USPS dataset #2

Open deep0learning opened 6 years ago

deep0learning commented 6 years ago

My data loading code is given below. I got the following error. Can you help to sort out please?

transform = transforms.Compose([ transforms.Resize((28,28)), transforms.Grayscale(num_output_channels=1), transforms.ToTensor(),
transforms.Normalize((0.1307,), (0.3081,))])

train_dataset = datasets.MNIST('/media/user/DATA/New_CODE/Working/CapsNet_pytorch/MNIST', download=True, transform=transform, train=True) test_dataset = USPS('/media/user/DATA/New_CODE/Working/CapsNet_pytorch/USPS', train=False, transform=transform, download=False)

train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=cfg.batch_size, shuffle=True)
test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=cfg.batch_size, shuffle=False)

train_batch_num = len(train_loader) test_batch_num = len(test_loader)

ERROR File "main.py", line 145, in predict, reconstruct_img = net(img_batch, train=False) File "/home/user/pytorch_python3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 224, in call result = self.forward(*input, *kwargs) File "/media/user/DATA/New_CODE/Working/CapsNet_pytorch/lib/network.py", line 41, in forward output = self.conv1(x) File "/home/user/pytorch_python3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 224, in call result = self.forward(input, **kwargs) File "/home/user/pytorch_python3/lib/python3.5/site-packages/torch/nn/modules/conv.py", line 254, in forward self.padding, self.dilation, self.groups) File "/home/user/pytorch_python3/lib/python3.5/site-packages/torch/nn/functional.py", line 52, in conv2d return f(input, weight, bias) RuntimeError: Need input.size[1] == 1 but got 28 instead.