pytorch报错: invalid argument 0: Sizes of tensors must match except in dimension 0.

pytorch报错:

invalid argument 0: Sizes of tensors must match except in dimension 0. Got 13 and 0 in dimension 2 at /opt/conda/conda-bld/pytorch_1556653114079/work/aten/src/TH/generic/THTensor.cpp:711

 

这个错误是由于pytorch在使用dataloader时,传入的数据会自动整合成一个个batch,要求其中的元素必须是相同的size,我的错是在语义分割时,传入的label和image大小不对应(搞不清楚linux文件排列的规则。。。)

你可能感兴趣的:(Deep,Learning)