Is CTL + TPU combination possible ?

Open
#177 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
4/5
Estimated time
3-5 days
Newbie friendliness
25/100
Issue type
Feature
Clarity
Needs clarification
Activity status
Stale
Tech stack
gcp, python, tensorflow
Domain
cloud, machine-learning

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.

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 ?

Dominant language
Python
Stars
383
Forks
93
Avg merge
1d 3h
Merged PRs (30d)
1

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from tensorflow/cloud

All issues in tensorflow/cloud

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.