仅当TPU训练Keras模型时出现形状错误

Question

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

这是我的notebook。只是给我一个错误，提示激活功能的大小不合适。

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

如果花了3个小时进行调试，我将为您提供任何帮助。

Answer 1

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

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

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

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