keras训练出现IndexError: tuple index out of range

最新推荐文章于 2024-07-29 23:27:25 发布

莫叶何竹

最新推荐文章于 2024-07-29 23:27:25 发布

阅读量2.3k

点赞数

分类专栏：踩过的坑文章标签： keras AI

本文链接：https://blog.youkuaiyun.com/weixin_40779727/article/details/109235461

版权

踩过的坑专栏收录该内容

5 篇文章

订阅专栏

我报该错误是因为：

triplet_loss = Lambda(self.cal_triplet_loss, name="triplet")(feature)
sim_model.compile(optimizer=optimizer,
                  loss={
                       "triplet": lambda y_true, y_pred: y_pred,
                        "output": "sparse_categorical_crossentropy",},
                        metrics=["accuracy"])

模型编译时， “triplet”层的输出是1d的，而metrics的计算在源码中需要至少两个维度。解决方案是，给triplet层的输出加一个维度从（batch_size, ) →（batch_size, 1）。
另一个解决方案就是去掉metrics，自己实现训练过程的指标计算（继承 keras.callbacks.Callback，实现其中的on_epoch_end, on_batch_end）。