YOLOv3_PyTorch icon indicating copy to clipboard operation
YOLOv3_PyTorch copied to clipboard

my dataset,why?

Open skysunsun opened this issue 5 years ago • 1 comments

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 15 and 16 in dimension 2 at c:\a\w\1\s\tmp_conda_3.7_110206\conda\conda-bld\pytorch_1550401474361\work\aten\src\thc\generic/THCTensorMath.cu:83

skysunsun avatar Apr 28 '19 05:04 skysunsun

you probably have an uneven-ish image size that freaks up downsampling/upsampling functions. Just crop the bigger tensor to have 15 as dimension 2 and it would be ok I think.

robertofratello avatar Jul 26 '19 07:07 robertofratello