PyTorch Forums
[URGENT ! !]RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 128 and 0 in dimension 2 at /opt/conda/conda-bld/pytorch_1544199946412/work/aten/src/TH/generic/THTensorMoreMath.cpp:1333
Xiaoyu_Song
(Xiaoyu Song)
January 31, 2019, 7:00am
3
Hi, there is something wrong with your dataloader, I think you should check you data sizes.
show post in topic