Merge pull request #990 from ChenYuHo/patch-3

[NFC/TF] Set GPU index based on local_rank instead of rank
2021-11-04 14:54:04 +01:00 · 2021-11-04 14:54:04 +01:00 · 2ac7ceb853
parent 22d6621dcd a5f26eeef7
commit 2ac7ceb853
1 changed files with 1 additions and 1 deletions
--- a/TensorFlow/Recommendation/NCF/ncf.py
+++ b/TensorFlow/Recommendation/NCF/ncf.py
@ -227,7 +227,7 @@ def main():
    # Create and run Data Generator in a separate thread
    data_generator = DataGenerator(
        args.seed,
-        hvd.rank(),
+        hvd.local_rank(),
        nb_users,
        nb_items,
        neg_mat,