RuntimeError：无效参数0：张量的大小必须匹配，但维0除外。维1为3和1。

Question

通过pytorch训练resnet50模型时，出现此错误：

  

RuntimeError：无效参数0：张量的大小必须匹配，除了   在维度0中。在维度1中有3和1   /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333

我正在使用：http://github.com/Helias/Car-Model-Recognition/

使用此数据集http://vmmrdb.cecsresearch.org/

Answer 1

我解决了这个问题，问题是图像的颜色通道不同，并非所有图像都是RGB，所以我在dataset.py中进行了转换，我对此进行了更改：

im = Image.open(image_path)

对此：

im = Image.open(image_path).convert('RGB')