Ajusta un modelo

En esta página, se muestra cómo interactuar con Vertex AI para ajustar un modelo con el panel lateral en Colab Enterprise.

Para acceder a los servicios y las APIs de Google Cloud ejecutando código en tu notebook de Colab Enterprise, puedes usar las credenciales asociadas con tu Cuenta de Google. Para obtener más información, consulta Cómo acceder a Google Cloud servicios y APIs.

El panel lateral es una forma adicional de interactuar con Vertex AI para ajustar modelos sin salir de la interfaz de Colab Enterprise.

El panel lateral aparece a la derecha de un notebook abierto.

Ajuste de modelos en Vertex AI

El ajuste de modelos es una forma eficaz de personalizar modelos grandes para tus tareas. Es un paso clave para mejorar la calidad y la eficiencia del modelo. El ajuste de modelos proporciona los siguientes beneficios:

  • Mayor calidad para tus tareas específicas
  • Mayor robustez del modelo
  • Menor latencia de inferencia y costo debido a instrucciones más cortas

Para obtener una descripción general del ajuste de modelos de Vertex AI para Gemini, consulta Introducción al ajuste.

Antes de comenzar

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  3. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

  5. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  6. Make sure that billing is enabled for your Google Cloud project.

  7. Enable the Vertex AI, Dataform, and Compute Engine APIs.

    Enable the APIs

Roles obligatorios

Para garantizar que tu cuenta de usuario tenga los permisos necesarios para usar el panel lateral en un notebook de Colab Enterprise, pídele a tu administrador que le otorgue a tu cuenta de usuario los siguientes roles de IAM en el proyecto:

Para obtener más información sobre cómo otorgar roles, consulta Administra el acceso a proyectos, carpetas y organizaciones.

Es posible que tu administrador también pueda otorgar a tu cuenta de usuario los permisos necesarios a través de roles personalizados o de otros roles predefinidos.

Ajusta un modelo

Puedes ajustar un modelo en Vertex AI con el panel lateral de Colab Enterprise.

  1. En la consola de Google Cloud, ve a la página Mis notebooks de Colab Enterprise.

    Ir a Mis notebooks

  2. En el menú Región, selecciona la región que contiene el notebook.

  3. Haz clic en el notebook que deseas abrir. Si aún no creaste un notebook, crea uno.

  4. A la derecha de la notebook, en el panel lateral, haz clic en el botón  Ajuste.

    El panel lateral expande la pestaña Ajuste.

  5. Haz clic en el botón Ajustar un modelo de Gemini.

    Colab Enterprise agrega celdas de código a tu notebook para ajustar un modelo de Gemini.

  6. En tu notebook, busca la celda de código que almacena los valores de los parámetros. Usarás estos parámetros para interactuar con Vertex AI.

  7. Actualiza los valores de los siguientes parámetros:

    • PROJECT_ID: Es el ID del proyecto en el que se encuentra tu notebook.
    • REGION: Es la región en la que se encuentra tu notebook.
    • TUNED_MODEL_DISPLAY_NAME: El nombre del modelo ajustado.
  8. En la siguiente celda de código, actualiza los parámetros de ajuste del modelo:

    • source_model: El modelo de Gemini que deseas usar, por ejemplo, gemini-1.0-pro-002.
    • train_dataset: Es la URL de tu conjunto de datos de entrenamiento.
    • validation_dataset: Es la URL de tu conjunto de datos de validación.
    • Ajusta los parámetros restantes según sea necesario.
  9. Ejecuta las celdas de código que el panel lateral agregó a tu notebook.

  10. Después de que se ejecute la última celda de código, haz clic en el botón  Ver trabajo de optimización que aparece.

  11. En el panel lateral, se muestra información sobre tu trabajo de ajuste de modelos.

    • La pestaña Supervisar muestra las métricas de ajuste cuando estas están listas.
    • En la pestaña Conjunto de datos, se muestra un resumen y métricas sobre el conjunto de datos después de que se procesa.
    • En la pestaña Detalles, se muestra información sobre tu trabajo de ajuste, como el método de ajuste y el modelo base (modelo de origen) que usaste.
  12. Una vez que se complete el trabajo de ajuste, puedes ir directamente de la pestaña Detalles de ajuste a una página en la que puedes probar tu modelo. Haz clic en Probar.

    La consola de Google Cloud se abre en la página Text chat de Vertex AI, donde puedes probar tu modelo.

Cómo ver los detalles de un trabajo de ajuste

Puedes ver los detalles de un trabajo de ajuste, incluso si no lo creaste ni lo ejecutaste en Colab Enterprise. Puedes ver los detalles de un trabajo de ajuste con el panel lateral de Colab Enterprise.

  1. En la consola de Google Cloud, ve a la página Mis notebooks de Colab Enterprise.

    Ir a Mis notebooks

  2. En el menú Región, selecciona la región que contiene el notebook.

  3. Haz clic en el notebook que deseas abrir. Si aún no creaste un notebook, crea uno.

  4. A la derecha de la notebook, en el panel lateral, haz clic en el botón  Ajuste.

    El panel lateral expande la pestaña Ajuste.

  5. Haz clic en el botón Ver detalles del trabajo de ajuste.

    Colab Enterprise agrega celdas de código a tu notebook para obtener los detalles de un trabajo de ajuste.

  6. En tu notebook, busca la celda de código que almacena los valores de los parámetros. Usarás estos parámetros para interactuar con Vertex AI.

  7. Actualiza los valores de los siguientes parámetros:

    • PROJECT_ID: Es el ID del proyecto en el que se encuentra tu notebook.
    • REGION: Es la región en la que se encuentra tu notebook.
    • TUNING_JOB_ID: Es el ID de tu trabajo de ajuste de Vertex AI.
  8. Ejecuta las celdas de código que el panel lateral agregó a tu notebook.

  9. Después de que se ejecute la última celda de código, haz clic en el botón  Ver trabajo de optimización que aparece.

  10. En el panel lateral, se muestra información sobre tu trabajo de ajuste de modelos.

    • La pestaña Supervisar muestra las métricas de ajuste cuando estas están listas.
    • En la pestaña Conjunto de datos, se muestra un resumen y métricas sobre el conjunto de datos después de que se procesa.
    • En la pestaña Detalles, se muestra información sobre tu trabajo de ajuste, como el método de ajuste y el modelo base (modelo de origen) que usaste.

¿Qué sigue?