maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Set GPU index based on local_rank instead of rank

Browse source
This commit is contained in:
Elton Chen-Yu Ho 2021-08-22 17:45:54 +08:00 committed by GitHub
parent ef98b2cef9
commit a5f26eeef7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
TensorFlow/Recommendation/NCF/ncf.py
View file

@ -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,