RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 736 and 129

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 736 and 1296 in dimension 2 at C:\w\1\s\tmp_conda_3.7_055457\conda\conda-bld\pytorch_1565416617654\work\aten\src\TH/generic/THTensor.cpp:689

在测试验证机时，由于验证集的图片大小都是不一样的，会报以上的错误。需要将batch_size设置为1，然而这样会导致模型只能一张张的预测，FPS下降非常快。最好将同尺寸的数据作为验证集，然后就可以使用batch_size来测试。