Skip to content

Add remote tensorflow profiling to training jobs. #519

Closed
@mkovalski

Description

To support on demand profiling of tensorflow , add handlers for profiling requests when running training jobs.

  • User updates their code to import the training util for profiling.
  • User submits a training job to Vertex AI.
  • Through Vertex AI Tensorboard, user can profile their remote training job through the "Capture Profile" functionality in Tensorboard.

Metadata

Assignees

Labels

api: aiplatformIssues related to the AI Platform API.type: feature request‘Nice-to-have’ improvement, new feature or different behavior or design.

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions