Я пытаюсь использовать больше ядер TPU для обучения. Мой исходный код использует keras для создания модели и keras_to_tpu_model для преобразования модели для работы на ядре TPU. Но мне нужно снизить скорость и повысить производительность. Кажется, я не могу изменить номер хоста или используемые ядра через какой-либо модуль. Как я могу это сделать?