Я пытаюсь запустить GridSearchCV на TPU в Google Compute Engine. Я могу нормально тренироваться с моделью Keras, но как только я запускаю ее через KerasClassifier и GridSearchCV, я получаю ошибку:
ValueError: Количество выборок 123221 не делится на размер партии 512.
Мой оригинальный X_train имеет размер 185005, затем я изменяю его размер, чтобы он соответствовал размеру batch_size - тогда он равен 184832.
Итак, в сообщении об ошибке указано 147865 сэмплов - я не понимаю, как оно там заканчивается.
Может ли кто-нибудь предоставить руководство по работе KerasClassifier и GridSearchCV на TPU?