googleapis / googleapis/python-aiplatform

Support Text summary in Experiment TensorBoard

Abierto
#6,338 2 comentarios 1 reacción 0 asignados Ver en GitHub
api: vertex-ai
Lenguaje dominante
Python
Estrellas
905
Forks
465
Merge medio
1 d 13 h
PR fusionados (30 d)
44

Descripción

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.

Guía de contribución

Abrir la guía de contribución

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.