invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1
原因:dataloader 需要的数据维度是一样的,有的有两个目标,有的有一个目标,就会出现异常:
解决方法:凑成一样的。
Traceback (most recent call last):
File "G:/project/facebox/faceboxes/tools/convert_tenser.py", line 116, in <module>
gen_data(batch_size,data_path,target_path)
File "G:/project/facebox/faceboxes/tools/convert_tenser.py", line 30, in gen_data
for step, samples in enumerate(dataloader):
File "D:\ProgramData\Miniconda3\lib\site-packages\torch\utils\data\dataloader.py", line 314, in __next__
batch = self.collate_fn([self.dataset[i] for i in indices])
File "D:\ProgramData\Miniconda