fendouai / PyTorchDocs

PyTorch 官方中文教程包含 60 分钟快速入门教程,强化教程,计算机视觉,自然语言处理,生成对抗网络,强化学习。欢迎 Star,Fork!

Home Page:http://pytorchchina.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

York1996OutLook opened this issue · 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

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