PyTorchDocs icon indicating copy to clipboard operation
PyTorchDocs copied to clipboard

batchsize=2的时候,连个图片的GTbox的数量不一样,拼接成tensor的时候就会报错

Open York1996OutLook opened this issue 6 years ago • 1 comments

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 5 and 4 in dimension 1 at ..\aten\src\TH/generic/THTensor.cpp:711

York1996OutLook avatar Nov 07 '19 11:11 York1996OutLook

确认一下 pytorch 的版本,查看一下函数的文档,看下是否有变化。

fendouai avatar Dec 05 '19 06:12 fendouai