search

BobLiu20 / YOLOv3_PyTorch

Full implementation of YOLOv3 in PyTorch
624 stars 191 forks source link

my dataset，why? #45

Open skysunsun opened 5 years ago

skysunsun commented 5 years ago

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

robertofratello commented 5 years ago

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.