tensorflow / recommenders-addons

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

Apache License 2.0

596 stars 136 forks source link

[FIX] horovod example and gpu unit test #442

Closed jq closed 5 months ago

jq commented 5 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

add inference
fix GPU unit tests

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: