Merge pull request #990 from ChenYuHo/patch-3

nv-kkudrynski · web-flow · commit 2ac7ceb8530e · 2021-11-04T14:54:04.000+01:00
[NFC/TF] Set GPU index based on local_rank instead of rank
diff --git 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,
