Konfigurasi skrip pelatihan Anda

Skrip pelatihan Anda harus dikonfigurasi untuk menulis log TensorBoard. Bagi pengguna TensorBoard yang sudah ada, tidak memerlukan perubahan pada kode pelatihan model Anda.

Untuk mengonfigurasi skrip pelatihan Anda di TensorFlow 2.x, buat callback TensorBoard dan tetapkan variabel log_dir ke lokasi mana pun yang dapat terhubung ke Google Cloud.

Callback TensorBoard kemudian disertakan dalam daftar callback model.fit TensorFlow.

import tensorflow as tf

def train_tensorflow_model_with_tensorboard(log_dir):
    (x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data()
    x_train, x_test = x_train / 255.0, x_test / 255.0

    def create_model():
        return tf.keras.models.Sequential(
            [
                tf.keras.layers.Flatten(input_shape=(28, 28)),
                tf.keras.layers.Dense(512, activation="relu"),
            ]
        )

    model = create_model()
    model.compile(
        optimizer="adam",
        loss="sparse_categorical_crossentropy",
        metrics=["accuracy"]
    )

    tensorboard_callback = tf.keras.callbacks.TensorBoard(
        log_dir=log_dir,
        histogram_freq=1
    )

    model.fit(
        x=x_train,
        y=y_train,
        epochs=5,
        validation_data=(x_test, y_test),
        callbacks=[tensorboard_callback],
    )

Log TensorBoard dibuat di direktori yang ditentukan dan dapat diupload ke eksperimen Vertex AI TensorBoard dengan mengikuti petunjuk Upload TensorBoard Logs untuk mengupload.

Untuk contoh lainnya, lihat dokumen open source TensorBoard

Langkah selanjutnya