In this tutorial, you learn how to enable Profiler in Vertex AI for custom training jobs with a prebuilt container.
Notebook: Profile model training performance using Cloud Profiler in prebuilt container
This tutorial uses the following Google Cloud ML services and resources:
- Vertex AI training
- Vertex AI TensorBoard
The steps performed include:
- Prepare your custom training code and load your training code as a Python package to a prebuilt container.
- Create and run a custom training job that enables Profiler.
- View the Profiler dashboard to debug your model training performance.