Is CTL + TPU combination possible ?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 383
- Forks
- 93
- Avg merge
- 1d 3h
- Merged PRs (30d)
- 1
Description
Is it possible to use tensorflow-cloud, with custom training loop on TPU ? I have checked both CTL and TPU example both. The CTL example explicitly defines the distribution strategy inside the code, which is not possible for the TPU case as we don't the TPU address to use. The example to which demonstrate running code on TPU uses the keras.fit api only. I just want to confirm the current implementation doesn't support this use case.
If it supports the use case I mention, how can I achieve that ?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Review the CTL and TPU examples referenced in the issue, then trace how the TPU address and distribution strategy are configured. Determine whether custom training loops are supported with tensorflow-cloud on TPU and document the supported setup or the missing implementation needed.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- gcp, python, tensorflow
- Domain
- cloud, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100