batchsize=2的时候,连个图片的GTbox的数量不一样,拼接成tensor的时候就会报错
York1996OutLook opened this issue · comments
York1996OutLook commented
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
fendouai commented
确认一下 pytorch 的版本,查看一下函数的文档,看下是否有变化。