tensorflow / recommenders-addons

Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.

Apache License 2.0

587 stars 132 forks source link

[opt] set retry_on_failure to false for tf allocator #407

Closed LinGeLin closed 4 months ago

LinGeLin commented 4 months ago

Description

set retry_on_failure to false for tf allocator

Fixes # (issue)

Type of change

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

Checklist:

[ ] I've properly formatted my code according to the guidelines
- [ ] By running yapf
- [ ] By running clang-format
[ ] This PR addresses an already submitted issue for TensorFlow Recommenders-Addons
[ ] I have made corresponding changes to the documentation
[ ] I have added tests that prove my fix is effective or that my feature works

How Has This Been Tested?

If you're adding a bugfix or new feature please describe the tests that you ran to verify your changes: