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 horovod example #440

Closed jq closed 2 months ago

jq commented 2 months ago

Description

fix load model bug
remove un-used self.gpu_device self.cpu_device
enable
```
sh start.sh 1
```
to test only 1 steps_per_epoch

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: