Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[tfjs-node] fixed summary writer memory leak #7490

Merged
merged 2 commits into from
Mar 16, 2023
Merged

Conversation

pyu10055
Copy link
Collaborator

@pyu10055 pyu10055 commented Mar 16, 2023

This is due to the step Int64Scalar internal tensor not deleted after use.
fixed #7432
also fixed a similar error occured in saved model execution memory leak when the input is of UINT8 or UINT64 dtype.
fixed #6937

To see the logs from the Cloud Build CI, please join either our discussion or announcement mailing list.


This change is Reviewable

Copy link
Collaborator

@Linchenn Linchenn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@pyu10055 pyu10055 merged commit 2a4c873 into master Mar 16, 2023
@pyu10055 pyu10055 deleted the summary_writer branch March 16, 2023 22:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

SummaryFileWriter memory leak? Memory Leak on tf.GraphModel.predict
3 participants