Skip to content

feat: pre batch creating TensorboardRuns and TensorboardTimeSeries in one_shot mode to speed up uploading - #772

Merged
yfang1 merged 2 commits into
googleapis:mainfrom
yfang1:batch_create
Oct 21, 2021
Merged

yfang1 merged 2 commits into
googleapis:mainfrom
yfang1:batch_create

Conversation

@yfang1

@yfang1 yfang1 commented Oct 14, 2021

Copy link
Copy Markdown
Contributor

The existing uploader iterates through the log dir and creates TensorboardRun and TensorboardTimeSeries on demand, one at a time, which slows down the uploading quite a bit.

This PR seeks to improve the one_shot performance by iterating through the log dir first, collects the Runs and TimeSeries ahead of time, batch create ahead of time so fewer RPCs are sent.

@yfang1
yfang1 requested a review from sasha-gitg October 14, 2021 21:08
@google-cla google-cla Bot added the cla: yes This human has signed the Contributor License Agreement. label Oct 14, 2021
@product-auto-label product-auto-label Bot added the api: aiplatform Issues related to the AI Platform API. label Oct 14, 2021
@karthikv2k
karthikv2k self-requested a review October 14, 2021 23:35
Comment thread google/cloud/aiplatform/tensorboard/uploader.py Outdated
Comment thread tests/unit/aiplatform/test_uploader.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader.py
Comment thread google/cloud/aiplatform/tensorboard/uploader.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader.py
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py Outdated
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py
Comment thread google/cloud/aiplatform/tensorboard/uploader_utils.py Outdated
@yfang1
yfang1 merged commit c9f68c6 into googleapis:main Oct 21, 2021
@yfang1
yfang1 deleted the batch_create branch October 21, 2021 04:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

api: aiplatform Issues related to the AI Platform API. cla: yes This human has signed the Contributor License Agreement.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants