深度学习，下载了github上的代码，换成自己的数据集后，遇到了这个问题要怎么解决呀？
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 16 and 1 in dimension 1 at C:\w\1\s\tmp_conda_3.7_100118\con
da\conda-bld\pytorch_1579082551706\work\aten\src\TH/generic/THTensor.cpp:612
应该是图片的尺寸有问题，和数据集对比一下