僅當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個小時的調試時間才能獲得任何幫助。

Answer 1

由於您是單標簽編碼，因此標簽不稀疏，因此需要使用“ categorical_accuracy”作為指標：

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

或更簡潔地使用“准確性”讓Keras根據所使用的損失函數（在這種情況下為'categorical_accuracy'推斷正確的度量標准，因為您將categorical_crossentropy用作損失函數）：

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