使用迁移学习训练Resnet-50时，如何解决此错误？

Question

我正在使用keras。版本“ 2.3.1”通过迁移学习来训练Resnet-50。

运行以下代码时出现错误：

history = model.fit_generator(generator=train_generator,
                              epochs= 3,
                              validation_data=validation_generator)

我得到的错误是：

Epoch 1/3
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-22-a42d5d67d1c7> in <module>
      1 history = model.fit_generator(generator=train_generator,
      2                               epochs= 3,
----> 3                               validation_data=validation_generator)
      4 
      5 

ValueError: Error when checking target: expected dense_2 to have shape (2,) but got array with shape (9,)

任何人都知道我该怎么解决？

Answer 1

这仅仅是因为模型输出形状与您的训练y（目标文件）不匹配。提供完整的脚本，例如您的训练数据的model_summary和形状（x，y）