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

FIX should set synchronous=True for TFRA optimizer #449

Closed jq closed 1 month ago

jq commented 1 month ago

Description

1 . should set synchronous=True for TFRA optimizer

should not use bp_v2 for sync training
fix the embedding size not div by mpi_size issue Brief Description of the PR:

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: