Ошибка формы только при тренировке ТПУ модели Keras

Question

Во-первых, это не мой код.Я просто изменил его, чтобы иметь возможность обучать его на ТПУ.Оригинальный автор здесь .Я могу запустить его на GPU-ускоренной среде выполнения на совместном ноутбуке, но кажется, что он сломается, когда я использую TPU-ускоренную среду выполнения.

Вот мой блокнот .Это просто дает мне ошибку, что функция активации имеет неправильный размер.

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

Я был бы признателен за любую помощь, которую я могу получить, потратив на отладку 3 часа.

Answer 1

Поскольку вы кодируете ярлыки одним щелчком и, следовательно, они не разрежены, вам нужно использовать «категорическую точность» в качестве метрики:

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

или более кратко используйте «точность», чтобы позволить Keras сделать выводправильная метрика, основанная на используемой функции потерь (которая в этом случае будет 'categorical_accuracy', поскольку вы используете categorical_crossentropy в качестве функции потерь):

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