RuntimeError: shape '[1024, 832, 1]' is invalid for input of size 3407872

[Liujiaze]

warnings.warn(warning.format(ret)) 0%| | 0/300 [00:00<?, ?it/s] Traceback (most recent call last): File "G:/STUDY CODE/CSRNet/CSRNet-pytorch-master/train.py", line 29, in for i, data in enumerate(tqdm(train_dataloader)): File "D:\Anaconda3\envs\pytorch\lib\site-packages\tqdm\std.py", line 1129, in iter for obj in iterable: File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 615, in next batch = self.collate_fn([self.dataset[i] for i in indices]) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 615, in batch = self.collate_fn([self.dataset[i] for i in indices]) File "G:\STUDY CODE\CSRNet\CSRNet-pytorch-master\dataset.py", line 50, in getitem dmap = self.dmap_transform(dmap) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torchvision\transforms\transforms.py", line 61, in call return F.to_tensor(pic) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torchvision\transforms\functional.py", line 71, in to_tensor img = img.view(pic.size[1], pic.size[0], nchannel) RuntimeError: shape '[1024, 832, 1]' is invalid for input of size 3407872

When I try to run train.py,I met the Error above,I can't found the reason,could you please help me.