-
Notifications
You must be signed in to change notification settings - Fork 428
Open
Labels
api: vertex-aiIssues related to the googleapis/python-aiplatform API.Issues related to the googleapis/python-aiplatform API.
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 theAIP_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:
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.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
api: vertex-aiIssues related to the googleapis/python-aiplatform API.Issues related to the googleapis/python-aiplatform API.