I am attaching full stack trace here

File “/media/kowshik/Data1/gan_expts/autoencoder/ae_mnisyt+usps.py”, line 129, in 
for count,data in enumerate(mnist_usps_dataloader):
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 286, in next
return self._process_next_batch(batch)
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 307, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 57, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 138, in default_collate
return [default_collate(samples) for samples in transposed]
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 138, in 
return [default_collate(samples) for samples in transposed]
File “/home/kowshik/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py”, line 115, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 1 and 2 at /opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/TH/generic/THTensorMath.c:3577