stack expects each tensor to be equal size

[Uzma Ahmed Din]

I got this error while running the train.py. How can i get the equal sizes?

File "/home/uzma/.local/lib/python3.6/site-packages/torch/_utils.py", line 434, in reraise
raise exception
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/uzma/.local/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/home/uzma/.local/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)
File "/home/uzma/TR-MISR/src/utils.py", line 65, in call
return self.collateFunction(batch)
File "/home/uzma/TR-MISR/src/utils.py", line 114, in collateFunction
padded_lr_batch = torch.stack(lr_batch, dim=0)
RuntimeError: stack expects each tensor to be equal size, but got [24, 32, 32] at entry 0 and [24, 32, 64] at entry 1