Custom training autologging: Notebook

As a data scientist experimenting with large models, you need a way to run experiments on a scalable training service to log parameters and metrics. This enables reproducibility.

With Gemini Enterprise Agent Platform training and experiments autologging integration, you can run your ML experiments at scale and autolog their parameters and metrics by using the enable_autolog argument.

Notebook: Vertex AI Experiments: Custom training autologging - Local script

This tutorial uses the following Cloud de Confiance ML services and resources:

  • Vertex AI Experiments
  • Gemini Enterprise Agent Platform training

The steps performed include:

  1. Formalize model experiment in a script.
  2. Run model training using local script on Gemini Enterprise Agent Platform training.
  3. Check out ML experiment parameters and metrics in Vertex AI Experiments.

Relevant content