search

MhLiao / DB

A PyTorch implementation of "Real-time Scene Text Detection with Differentiable Binarization".
2.11k stars 481 forks source link

Sizes of tensors must match except in dimension 0. Got 2 and 1 in dimension 1 #374

Open sssimpleboy opened 1 year ago

sssimpleboy commented 1 year ago

作者您好，我在验证阶段，我的数据集打包失败，请问应该怎么办呢？ 我试过只打包图片，可以成功，但是后续计算损失之类的，我还没改 1692767494474

sssimpleboy commented 1 year ago

只要验证时候batch设的大于1就会报错，所以我现在只能设成1 . 但是设成1的话，模型就不能多卡验证。。 所以我就自己重写了个validate 去单卡验证，但是这样又不能 边训练边验证。。 甚是苦恼 。