python - 仅在 TPU 训练 Keras 模型时出现形状错误

Question

首先，这不是我的代码。我只是将其更改为能够在 TPU 上进行训练。原作者在这里。我能够在协作笔记本上的 GPU 加速运行时运行它，但是当我执行 TPU 加速运行时它似乎中断了。

这是我的笔记本。它只是给我一个错误，即激活函数的大小不正确。

ValueError: Error when checking target: expected activation_21 to have shape (1,) but got array with shape (205,)

如果我花了大约 3 个小时进行调试，我将不胜感激。

score 0 · Accepted Answer

由于您对标签进行一次性编码，因此它们不是稀疏的，因此您需要使用“categorical_accuracy”作为指标：

model.compile(..., metrics=['categorical_accuracy'])

或更简洁地使用“准确性”让 Keras 根据使用的损失函数推断正确的度量（在这种情况下，'categorical_accuracy'因为您使用categorical_crossentropy的是损失函数）：

model.compile(..., metrics=['accuracy'])

1 回答 1