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

[sssimpleboy]

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

[sssimpleboy]

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