首页 > 解决方案 > 在 keras 上编译模型时出现索引错误

问题描述

我试图在本地文件夹中的一系列图像上训练一个非常简单的神经网络,但是当我尝试编译我的模型时出现错误。

这是我的代码

model.compile(optimizer=keras.optimizers.Adam(Lr), 
              loss={'centernet_loss': lambda y_true, y_pred: y_pred}, 
              metrics=['accuracy']) 

当我运行它时,我得到了这个错误:

IndexError:
list index out of range. 

当我将代码更改为此

model.compile(optimizer=keras.optimizers.Adam(Lr), 
              loss={'centernet_loss': lambda y_true, y_pred: y_pred}) 

它可以正常运行。你知道怎么解决吗?

标签: kerasmetrics

解决方案


推荐阅读