RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension

可神奇了本来bantchsize设置到20都没问题，就是gpu利用率上不去，一会等于2也可以运行，一会2也不行。无奈，网上说设置成图片数量的因数，也没用，换了个数据集。可以了
可能是图片通道问题吗，