Guía de rendimiento de TPU de Cloud
El primer paso para solucionar problemas de rendimiento de la TPU es crear un perfil de tu modelo. Para obtener más información sobre cómo capturar un perfil de rendimiento, consulta el artículo Elaborar un perfil de modelo en TPU de Cloud.
Rendimiento del modelo de TPU
En esta sección se describen los problemas generales que pueden reducir el rendimiento del modelo y cómo puedes solucionarlos.
El modelo está limitado por la entrada
Las TPUs realizan cálculos muy rápido. Para asegurarte de que la TPU no esté inactiva, es importante que se cargue un flujo constante de datos en ella. La forma de hacerlo depende de cómo cargues y preproceses tu conjunto de datos. Por ejemplo, puedes leer archivos de datos en paralelo con tf.data.TFRecordset() y el parámetro