Open ReyonRen opened 3 years ago
Hi all, thanks for the cool contribution.
I noticed that TPUEstimator is used in the code. After running on a machine with multiple GPUs, I found that only one GPU is in use. How can I modify the code so that the code can run in parallel on multiple GPUs?
TPUEstimator
Thank you!
Hi all, thanks for the cool contribution.
I noticed that
TPUEstimator
is used in the code. After running on a machine with multiple GPUs, I found that only one GPU is in use. How can I modify the code so that the code can run in parallel on multiple GPUs?Thank you!