Skip to content
This repository has been archived by the owner on Jun 18, 2024. It is now read-only.

Finetune with multiple GPU #228

Open
ReyonRen opened this issue Sep 11, 2020 · 0 comments
Open

Finetune with multiple GPU #228

ReyonRen opened this issue Sep 11, 2020 · 0 comments

Comments

@ReyonRen
Copy link

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!
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ReyonRen