Skip to content

Support Text summary in Experiment TensorBoard #6338

@LouisChirol

Description

@LouisChirol

Hello,

I'm migrating some ML tasks to Vertex AI and really like the Experiment framework.

My current setup is:

  • submitting CustomJob tasks from a custom training image
  • group these tasks in an Experiment, to enable an easy comparison between those
  • the training script is adapted to log information both to Experiment (with the Python SDK, e.g. log_time_series_metrics()) and to the CustomJob TB (standard SummaryWriter in the AIP_TENSORBOARD_LOG_DIR), following the documentation

It's all very convenient, but there is one thing I cannot do with Experiment.
The Python SDK does not support text uploads (and hparams but that's not crucial), and it's very important for NLP or ASR training tasks.
This is how it looks like in my CustomJob TB:

Image

I'm opening this feature request as this would be a handsome addition to the Experiment framework.

Thank you for your work, don't hesitate if I can provide more input.

Metadata

Metadata

Assignees

No one assigned

    Labels

    api: vertex-aiIssues related to the googleapis/python-aiplatform API.

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions