huggingface / huggingface/accelerate
Can accelerate train a single model on multiple TPU VMs (not a TPU Pod)?
Open
feature request
TPU
- Dominant language
- Python
- Stars
- 9.9k
- Forks
- 1.5k
- Avg merge
- 5d 2h
- Merged PRs (30d)
- 27
Description
I know accelerate supports training on a TPU Pod( #1049 and #471 ), but I only have **5 TPU v3-8 VMs** obtained from the TRC program. I wanted to use all these 5 VMs to train a single model, but I could not find an appropriate way to make it work by using accelerate.
Can someone kindly tell me how I can make this work, or is it just not possible?
Contributor guide
Assessment
This issue has not been assessed yet.