Hi, I'm looking to train GPT-2 in Spanish. And I would like to know exactly what hardware is needed.
On the readme you state that was trained Using Cloud TPU Pod v3-256 to train 22w steps.
- Are you referring to the following TPU POD?
| TPU (v3) type |
TPU v3 cores |
TPU total memory |
Available zones |
| v3-256 |
256 |
4 TiB |
europe-west4-a |
Thank you!