googleapis / googleapis/python-aiplatform

Support Text summary in Experiment TensorBoard

Open
#6,338 2 comments 1 reaction 0 assignees View on GitHub
api: vertex-ai
Dominant language
Python
Stars
905
Forks
465
Avg merge
1d 13h
Merged PRs (30d)
44

Description

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.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.