tensorflow / addons

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

Apache License 2.0

1.69k stars 611 forks source link

Fix Cutout memory leak #2815

Closed drauh closed 1 year ago

drauh commented 1 year ago

Description

Fixes #2814

Type of change

[x] Bug fix
[ ] New Tutorial
[ ] Updated or additional documentation
[ ] Additional Testing

Checklist:

[x] I've properly formatted my code according to the guidelines
- [x] By running Black + Flake8
- [x] By running pre-commit hooks
[x] This PR addresses an already submitted issue for TensorFlow Addons
[x] I have made corresponding changes to the documentation
[ ] I have added tests that prove my fix is effective or that my feature works
[ ] This PR contains modifications to C++ custom-ops

How Has This Been Tested?

By running the code from the issue reproducing the memory leak, here are the outputs now:

1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.32 GB
1.33 GB

bot-of-gabrieldemarmiesse commented 1 year ago

@fsx950223

You are owner of some files modified in this pull request. Would you kindly review the changes whenever you have the time to? Thank you very much.